
    5;ji;                       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
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_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.preview.trainingreturnintc                 J    t          j                    t                   d         S )z*Return current opset for domain `ai.onnx`.   )r   schema_version_mapr        L/root/voice-cloning/.venv/lib/python3.11/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%   0   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_valuer0   D   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;   N   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=))r3   type_strdescriptionoptionis_homogeneous	min_aritydifferentiation_categoryr*   s    r    _op_schema_formal_parameter_reprrE   _   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_strsr@   r*   s    r   %_op_schema_type_constraint_param_reprrI   k   sH    	[t7J 	[ 	[!3	[ 	[EIEU	[ 	[ 	[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>   )r3   typer@   default_valuerequiredr*   s    r   _op_schema_attribute_reprrN   u   sh    	M49 	M 	Mty 	M 	MRVRb 	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   rU   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 )zRegister 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   r4   r5   getminmaxset_domain_to_versionr	   )rU   version_mapr4   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   r1   )r   rO   )rU   r   r   rX   )&
__future__r   __all__r   onnx.onnx_cpp2py_export.defsonnx_cpp2py_exportdefsr   r   r   r   
has_schemar   r   r   r   r
   r   r   propertyr%   r   function_bodyr+   non_deterministicr0   	AttributerL   r;   __repr__rE   FormalParameterrI   TypeConstraintParamrN   r   r   r	   r   r   r   <module>ro      s   # " " " " "  "  ( ( ( ( ( ( ( ( ("<  l\
# = ' 2 2 2 2
5 5 5 5
 
  
 :(  
K K 
K
 0  
  
 $<       $     %E  !    )N  %    8      m     r   