§
    2�Åjw  ã                  ó²  — d dl mZ g d¢Zd dlZd dlmc mZ dZdZ	dZ
dZej        Zej        Zej        Zej        Zej        Zdd
„Zdd„Zed„ ¦   «         Zej        Zee_        ed„ ¦   «         Zee_        ed„ ¦   «         Zeej        _        dd„Zee_        dd„Zeej         _        dd„Z!e!ej"        _        dd„Z#e#ej        _        dd„Z$ej%        Z%dd„Z&dS )é    )Úannotations)ÚCÚONNX_DOMAINÚONNX_ML_DOMAINÚAI_ONNX_PREVIEW_DOMAINÚAI_ONNX_PREVIEW_TRAINING_DOMAINÚhasÚregister_schemaÚderegister_schemaÚ
get_schemaÚget_all_schemasÚget_all_schemas_with_historyÚonnx_opset_versionÚget_function_opsÚOpSchemaÚSchemaErrorNÚ z
ai.onnx.mlzai.onnx.previewzai.onnx.preview.trainingÚreturnÚintc                 óJ   — t          j        ¦   «         t                   d         S )z*Return current opset for domain `ai.onnx`.é   )r   Úschema_version_mapr   © ó    úCC:\Projects\Historical\venv\Lib\site-packages\onnx/defs/__init__.pyr   r   (   s   € åÔÑ!Ô!¥+Ô.¨qÔ1Ð1r   c                 óJ   — t          j        ¦   «         t                   d         S )z-Return current opset for domain `ai.onnx.ml`.r   )r   r   r   r   r   r   Úonnx_ml_opset_versionr   -   s   € åÔÑ!Ô!¥.Ô1°!Ô4Ð4r   c                ó`   — t          j        ¦   «         }|                     | j        ¦  «         |S ©N)ÚonnxÚFunctionProtoÚParseFromStringÚ_function_body)ÚselfÚ
func_protos     r   Ú_function_protor&   2   s-   € åÔ#Ñ%Ô%€JØ×Ò˜tÔ2Ñ3Ô3Ð3ØÐr   c                ó6   — | j         t          j        j        k    S )z+Check if the operator is non-deterministic.)Únode_determinismr   ÚNodeDeterminismÚDeterministic©r$   s    r   Ú_non_deterministicr,   =   s   € ð Ô ¥HÔ$<Ô$JÒJÐJr   c                ó`   — t          j        ¦   «         }|                     | j        ¦  «         |S r   )r    ÚAttributeProtor"   Ú_default_value)r$   Úattrs     r   Ú_attribute_default_valuer1   F   s,   € åÔÑ Ô €DØ×Ò˜Ô,Ñ-Ô-Ð-Ø€Kr   Ústrc                óˆ   — d| j         ›d| j        ›d| j        ›d| j        ›d| j        ›d| j        ›d| j        ›d| j        ›d	�S )
NzOpSchema(
    name=z,
    domain=z,
    since_version=z
,
    doc=z,
    type_constraints=z,
    inputs=z,
    outputs=z,
    attributes=z
))ÚnameÚdomainÚsince_versionÚdocÚtype_constraintsÚinputsÚoutputsÚ
attributesr+   s    r   Ú_op_schema_reprr<   P   s™   € ð
à
Œ)ð
ð 
ð ŒKð
ð 
ð Ô%ð	
ð 
ð
 
Œð
ð 
ð Ô+ð
ð 
ð ŒKð
ð 
ð Œ\ð
ð 
ð ”ð
ð 
ð 
ð 
r   c                óx   — d| j         ›d| j        ›d| j        ›d| j        ›d| j        ›d| j        ›d| j        ›d�S )	NzOpSchema.FormalParameter(name=z, type_str=ú, description=z, param_option=z, is_homogeneous=z, min_arity=z, differentiation_category=ú))r4   Útype_strÚdescriptionÚoptionÚis_homogeneousÚ	min_arityÚdifferentiation_categoryr+   s    r   Ú _op_schema_formal_parameter_reprrF   a   s�   € ð	G¨¬ð 	Gð 	GÀÄð 	Gð 	GØÔ'ð	Gð 	GØ:>¼+ð	Gð 	GàÔ-ð	Gð 	Gà=A¼^ð	Gð 	Gð %)Ô$Að	Gð 	Gð 	Gðr   c                ó8   — d| j         ›d| j        ›d| j        ›d�S )Nz,OpSchema.TypeConstraintParam(type_param_str=z, allowed_type_strs=r>   r?   )Útype_param_strÚallowed_type_strsrA   r+   s    r   Ú%_op_schema_type_constraint_param_reprrJ   m   sH   € ð	[°tÔ7Jð 	[ð 	[Ø!Ô3ð	[ð 	[ØEIÔEUð	[ð 	[ð 	[ðr   c                óX   — d| j         ›d| j        ›d| j        ›d| j        ›d| j        ›d�S )NzOpSchema.Attribute(name=z, type=r>   z, default_value=z, required=r?   )r4   ÚtyperA   Údefault_valueÚrequiredr+   s    r   Ú_op_schema_attribute_reprrO   w   sh   € ð	M 4¤9ð 	Mð 	M°t´yð 	Mð 	MÐRVÔRbð 	Mð 	MØÔ+ð	Mð 	MØ:>¼-ð	Mð 	Mð 	Mðr   úlist[OpSchema]c                 ó@   — t          j        ¦   «         } d„ | D ¦   «         S )z&Return operators defined as functions.c                ó.   — g | ]}|j         s|j        ¯|‘ŒS r   )Úhas_functionÚhas_context_dependent_function)Ú.0Úschemas     r   ú
<listcomp>z$get_function_ops.<locals>.<listcomp>„   s>   € ð ð ð àØÔðð #)Ô"GðØðð ð r   )r   r   )Úschemass    r   r   r   �   s1   € åÔÑ!Ô!€Gðð àðñ ô ð r   rV   r   ÚNonec                óB  — t          j        ¦   «         }| j        }| j        }|                     |||f¦  «        \  }}||vs||cxk    r|k    s=n t          ||¦  «        }t          ||¦  «        }t          j        | j        ||¦  «         t          j        | ¦  «         dS )z¼Register a user provided OpSchema.

    The function extends available operator set versions for the provided domain if necessary.

    Args:
        schema: The OpSchema to register.
    N)	r   r   r5   r6   ÚgetÚminÚmaxÚset_domain_to_versionr
   )rV   Úversion_mapr5   ÚversionÚmin_versionÚmax_versions         r   r
   r
   Ž   s³   € õ Ô&Ñ(Ô(€KØŒ]€FØÔ"€GØ*Ÿš¨v¸ÀÐ7IÑJÔJÑ€K�Ø�[Ð Ð ¨¸Ð)NÐ)NÒ)NÐ)NÀ;Ò)NÐ)NÐ)NÐ)NÝ˜+ wÑ/Ô/ˆÝ˜+ wÑ/Ô/ˆÝ	Ô ¤¨{¸KÑHÔHÐHÝÔ�fÑÔÐÐÐr   )r   r   )r   r2   )r   rP   )rV   r   r   rY   )'Ú
__future__r   Ú__all__r    Úonnx.onnx_cpp2py_export.defsÚonnx_cpp2py_exportÚdefsr   r   r   r   r   Ú
has_schemar	   r   r   r   r   r   r   Úpropertyr&   r   Úfunction_bodyr,   Únon_deterministicr1   Ú	AttributerM   r<   Ú__repr__rF   ÚFormalParameterrJ   ÚTypeConstraintParamrO   r   r   r
   r   r   r   ú<module>rp      sò  ðð #Ð "Ð "Ð "Ð "Ð "ðð ð €ð$ €€€Ø (Ð (Ð (Ð (Ð (Ð (Ð (Ð (Ð (à€Ø€Ø*Ð Ø"<Ð ð „l€ØŒ\€
ØÔ#€Ø Ô=Ð ØÔ'Ð ð2ð 2ð 2ð 2ð
5ð 5ð 5ð 5ð
 
ðð ñ 
„ðð Œ:€Ø(€Ô ð 
ðKð Kñ 
„ðKð
 0€Ô ð 
ðð ñ 
„ðð $<€Ô Ô  ðð ð ð ð $€Ô ðð ð ð ð %E€Ô Ô !ðð ð ð ð )N€Ô Ô %ðð ð ð ð 8€Ô Ô ðð ð ð ð Œm€ðð ð ð ð ð r   