2j dZddlmZddlmZmZddlZddlm Z erddl m Z ddl m Z mZGdd eZGd d ZddZddZdS)aEOptional runtime validation of static shape annotations. This module implements the runtime semantics described in ``docs/ShapeAnnotationSemantics.md``: each symbolic dimension name (``dim_param``) occurring in a model is treated as existentially quantified once per inference run. A :class:`SymbolBindings` instance is a partial map from such names to the (non-negative integer) value they were first observed to take during a run; every later occurrence of the same name is checked against that first binding. As a convenience, alongside shape checking this module also validates a tensor-typed value's element type (``elem_type``) against its declared ``TensorProto.DataType``, since this is a simple, purely local check (it does not involve symbolic dimensions or bindings) that ONNX's static checker already performs, but which is otherwise easy to violate undetected at runtime (e.g. a kernel producing the wrong dtype). This module only supports today's single, global namespace of symbolic dimension names (a name means the same thing everywhere it occurs in a model, including inside nested subgraphs) and only validates tensor-typed values. Locally scoped names (for example, a name local to one iteration of a ``Loop`` body, or to one element of a ``Sequence``) are not addressed; see ``docs/ShapeAnnotationSemantics.md`` for context. ) annotations) TYPE_CHECKINGAnyN)tensor_dtype_to_np_dtype)Sequence)TensorShapeProto TypeProtoceZdZdZdS)ShapeAnnotationErrorzYRaised when a value's actual runtime shape violates its declared static shape annotation.N)__name__ __module__ __qualname____doc__XC:\Projects\Historical\venv\Lib\site-packages\onnx/reference/shape_annotation_checker.pyr r +sccccrr c"eZdZdZd dZdd Zd S)SymbolBindingsaTracks the runtime value bound to each symbolic dimension name (``dim_param``) during one inference run. A single instance of this class should be used for exactly one call to :meth:`onnx.reference.ReferenceEvaluator.run`, since a symbolic dimension name is existentially quantified once per run: different runs of the same model may bind the same name to different values. returnNoneci|_dS)N) _bindings)selfs r__init__zSymbolBindings.__init__8s )+rdimTensorShapeProto.Dimensionactualintcontextstrc @|d}|dkr*|j|krt|d|jd|ddS|dkrP|j}|j|}| ||j|<dS||krt|d|d |d |ddSdS) aLChecks a single dimension against an actual (runtime) axis size, binding it if unbound. Args: dim: the declared ``Dimension`` (``dim_value``, ``dim_param``, or neither). actual: the actual runtime size of the corresponding axis. context: a human-readable description of the value and axis being checked, used only for error messages. Raises: ShapeAnnotationError: if ``dim`` has a ``dim_value`` that disagrees with ``actual``, or a ``dim_param`` already bound to a different value. value dim_valuez: declared dimension value z' does not match actual dimension value . dim_paramNz: symbolic dimension z was previously bound to z,, but this value has actual dimension value ) WhichOneofr#r r%rget)rrrrwhichnamebounds rcheck_and_bindzSymbolBindings.check_and_bind;sw'' K  }&&*883=88.4888'& k ! !=DN&&t,,E}'-t$$$&*KKTKKV[KKAGKKK " ! !rN)rr)rrrrrr rr)r r rrrr+rrrrr/sF,,,,rr shape_protoTensorShapeProto | None actual_shape Sequence[int]bindingsr)r rrc b|dSt|j}t|}||kr)td|d|d|dt|d t t |j|dD]3\}\}}||t|d|d | 4dS) a=Checks an actual runtime tensor shape against a declared static shape annotation. This implements the per-value check described in ``docs/ShapeAnnotationSemantics.md``: a missing ``shape_proto`` (unknown rank) is always satisfied; a rank mismatch always fails; otherwise each axis is checked (and, for symbolic dimensions, bound) via ``bindings``. Args: shape_proto: the declared ``TensorShapeProto``, or ``None`` if the value's type declares no shape (unknown rank). actual_shape: the actual runtime shape of the value. bindings: the :class:`SymbolBindings` for the current inference run. name: the name of the value being checked, used only for error messages. Raises: ShapeAnnotationError: if the actual shape does not satisfy the annotation. NValue z%: declared shape annotation has rank z , but the actual value has rank z (shape z).F)strictz, axis )r)lenrr tuple enumeratezipr+r) r,r.r0r) declared_rank actual_rankaxisrrs r check_shaper;_s. ((Ml##K ##" ZT Z Z- Z Z-8 Z ZBG BUBU Z Z Z    ) KO\%888    msF  V&Dt&D&Dd&D&D       r type_protoTypeProto | Noner"rc ||dsdSt|dd}|dS|j}|jrt|dd}|} t |j}n#t $rd}YnwxYw|t jtkot|dddk}|!||kr|std|d|d |d |dr|j nd} t| |||dS) a<Checks a runtime value against its declared ``TypeProto``, if it declares a tensor shape. Only tensor-typed values with a ``.shape`` attribute (e.g. a NumPy array) are checked in this initial implementation; other value kinds (sequences, maps, optionals, or a missing/absent type) are silently skipped. See the module docstring for the scope of what this checks. Args: type_proto: the value's declared ``TypeProto``, or ``None`` if not known. value: the actual runtime value produced during execution. bindings: the :class:`SymbolBindings` for the current inference run. name: the name of the value being checked, used only for error messages. Raises: ShapeAnnotationError: if the value's shape or element type does not match its declared type. N tensor_typeshapedtypekindUr2z: declared element type z$ does not match actual element type r$) HasFieldgetattrr? elem_typerKeyErrornprAobjectr r@r;) r<r"r0r)r@r? actual_dtypedeclared_dtypestring_unicoder,s rcheck_value_against_typerMsz.!4!4]!C!C E7D ) )E } (Kugt44  # &!9+:O!P!P & & &"& &"(6"2"22?L&$773>  * N22&3+ATAA^AA1=AAA(3';';G'D'DN+##$K UHd33333sA&& A54A5) r,r-r.r/r0rr)r rr) r<r=r"rr0rr)r rr)r __future__rtypingrrnumpyrH onnx.helperrcollections.abcr onnx.onnx_pbrr RuntimeErrorr rr;rMrrrrUs2#"""""%%%%%%%%0000009((((((88888888ddddd<ddd++++++++`% % % % P848484848484r