§
    Ë²Åjˆ  ã                   óp  — d Z ddlZddlZddlmZ ddlmZmZ ddlm	Z	m
Z
  ej        e
j        d¬¦  «        Z ed¦  «        d	e	j        fd
„¦   «         Z ed¦  «        ej        d	e	j        fd„¦   «         ¦   «         Z ed¦  «        dd	e	j        fd„¦   «         Z ed¦  «        d	e	j        fd„¦   «         ZdS )am  This file exports ONNX ops for opset 15.

Note [ONNX operators that are added/updated in opset 15]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
https://github.com/onnx/onnx/blob/master/docs/Changelog.md#version-15-of-the-default-onnx-operator-set
New operators:
    Bernoulli
    CastLike
    Optional
    OptionalGetElement
    OptionalHasElement

Updated operators:
    BatchNormalization https://github.com/onnx/onnx/pull/3545
                        Backwards compatible
                        TODO: test coverage for mixed types inputs.
    Pow                https://github.com/onnx/onnx/pull/3412
                        Backwards compatible
                        TODO: bfloat16 support.
    Shape              https://github.com/onnx/onnx/pull/3580
                        Backwards compatible
                        TODO: optional start/end attribute.
é    N)Ú_C)Úsymbolic_helperÚsymbolic_opset9)Ú	jit_utilsÚregistrationé   )Úopsetzaten::__is_Úgc                 óZ  — t          j        |¦  «        r‚t          |                     ¦   «         t          j        ¦  «        r,|                      d|¦  «        }|                      d|¦  «        S |                      dt          j        dg¦  «        ¬¦  «        S t          j
        | ||¦  «        S )NÚOptionalHasElementÚNotÚConstantr   )Úvalue_t)r   Ú_is_noneÚ
isinstanceÚtyper   ÚOptionalTypeÚopÚtorchÚ
BoolTensorÚopset9Úeq)r
   ÚselfÚotherÚnones       úLC:\Projects\Historical\venv\Lib\site-packages\torch/onnx/symbolic_opset15.pyÚ	aten__is_r   (   s’   € åÔ Ñ&Ô&ð CÝ�d—i’i‘k”k¥2¤?Ñ3Ô3ð 	CØ—4’4Ð,¨dÑ3Ô3ˆDØ—4’4˜˜tÑ$Ô$Ð$à—4’4˜
­EÔ,<¸a¸SÑ,AÔ,A�4ÑBÔBÐBÝŒ9�Q˜˜eÑ$Ô$Ð$ó    zaten::__isnot_c                 ó$   — t          | ||¦  «        S )N)r   )r
   r   r   s      r   Úaten__isnot_r    3   s   € õ �Q˜˜eÑ$Ô$Ð$r   zaten::bernoullic                 ó:  — |�*t          j        |¦  «        st          j        dd|¦  «         |�*t          j        |¦  «        st          j        dd|¦  «         |�t          j        |¦  «        r|                      d|¦  «        S t	          j        | ||||¦  «        S )NÚ	Bernoulliz,out parameter is not supported for bernoulliz(generator is not supported for bernoulli)r   r   Ú_unimplementedr   r   Ú	bernoulli)r
   ÚinputÚpÚ	generatorÚouts        r   r$   r$   9   s«   € à
€�Ô7¸Ñ<Ô<€ÝÔ&ØÐGÈñ	
ô 	
ð 	
ð Ð¥_Ô%=¸iÑ%HÔ%HÐÝÔ&ØÐCÀUñ	
ô 	
ð 	
ð 	€y•OÔ,¨QÑ/Ô/€yØ�tŠt�K Ñ'Ô'Ð'ÝÔ˜A˜u a¨°CÑ8Ô8Ð8r   zprim::unchecked_castc                 óŠ   — t          |                     ¦   «         t          j        ¦  «        r|                      d|¦  «        S |S )NÚOptionalGetElement)r   r   r   r   r   )r
   r   s     r   Úprim_unchecked_castr+   H   s:   € õ
 �$—)’)‘+”+�rœÑ/Ô/ð 0Ø�tŠtÐ(¨$Ñ/Ô/Ð/à€Kr   )NNN)Ú__doc__Ú	functoolsr   r   Ú
torch.onnxr   r   r   Útorch.onnx._internalr   r   ÚpartialÚonnx_symbolicÚ_onnx_symbolicÚGraphContextr   Úwrap_logical_op_with_negationr    r$   r+   © r   r   ú<module>r6      sx  ððð ð6 Ð Ð Ð à €€€Ø Ð Ð Ð Ð Ð Ø AÐ AÐ AÐ AÐ AÐ AÐ AÐ AØ 8Ð 8Ð 8Ð 8Ð 8Ð 8Ð 8Ð 8ð #�Ô" <Ô#=ÀRÐHÑHÔH€ð €�ÑÔð%�Ô'ð %ð %ð %ñ Ôð%ð €Ð Ñ!Ô!ØÔ%ð%�IÔ*ð %ð %ð %ñ &Ô%ñ "Ô!ð%ð €Ð!Ñ"Ô"ð9ð 9�Ô'ð 9ð 9ð 9ñ #Ô"ð9ð €Ð&Ñ'Ô'ð˜9Ô1ð ð ð ñ (Ô'ðð ð r   