
    
Ng,                        g d Z ddlmZ ddlmc mZ ddlmZm	Z	 dZ
dZdZej        Zej        Zej        Zej        Zej        Zdefd	Zdefd
Zed             Zej        Zee_        ed             Zeej        _        defdZee_        defdZ e ej!        _        defdZ"e"ej#        _        defdZ$e$ej        _        dee         fdZ%ej&        Z&deddfdZ'dS ))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SchemaError    )ListN)AttributeProtoFunctionProto z
ai.onnx.mlzai.onnx.preview.trainingreturnc                  J    t          j                    t                   d         S )z*Return current opset for domain `ai.onnx`.   )r   schema_version_mapr        N/var/www/html/ai-engine/env/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                 V    t                      }|                    | j                   |S N)r   ParseFromString_function_body)self
func_protos     r   _function_protor$   1   s(    Jt2333r   c                 V    t                      }|                    | j                   |S r   )r   r    _default_value)r"   attrs     r   _attribute_default_valuer(   <   s)    D,---Kr   c                     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
attributesr"   s    r   _op_schema_reprr3   F   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=))r*   type_strdescriptionoptionis_homogeneous	min_aritydifferentiation_categoryr2   s    r    _op_schema_formal_parameter_reprr=   W   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=r5   r6   )type_param_strallowed_type_strsr8   r2   s    r   %_op_schema_type_constraint_param_reprrA   c   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=r5   z, default_value=z, required=r6   )r*   typer8   default_valuerequiredr2   s    r   _op_schema_attribute_reprrF   m   sh    	M49 	M 	Mty 	M 	MRVRb 	M 	M+	M 	M:>-	M 	M 	Mr   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>z   s(    iiivF,?i6ChiFiiir   )r   r
   )schemass    r   r   r   w   s$    !!Giiiiiir   rL   c                 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   r+   r,   getminmaxset_domain_to_versionr   )rL   version_mapr+   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   )(__all__typingr   onnx.onnx_cpp2py_export.defsonnx_cpp2py_exportdefsr   onnxr   r   r   r   r   
has_schemar   r	   r
   r   r   intr   r   propertyr$   r   function_bodyr(   	AttributerD   strr3   __repr__r=   FormalParameterrA   TypeConstraintParamrF   r   r   r   r   r   r   <module>rg      s  
  "       ( ( ( ( ( ( ( ( ( . . . . . . . ."<  l\
# = ' 2C 2 2 2 2
5s 5 5 5 5
 
  
 :(  
  
 $<   S     $ c     %E  !3     )N  %s     8  j$x. j j j j mH       r   