o
    öXœi  ã                   @   s”  d Z g d¢ZddlZddlmZmZmZmZmZm	Z	m
Z
 ddlmZmZmZ ddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3 ddl4m5Z5m6Z6 ddl7m8Z8m9Z9 dd	l:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZK dd
lLmMZM ddlNmOZOmPZPmQZQmRZRmSZS dZTddd„ZUeUƒ  dZVdS )z6In-memory intermediate representation for ONNX graphs.)JÚserdeÚ	traversalÚconvenienceÚexternal_dataÚtapeÚschemasÚTensorÚExternalTensorÚStringTensorÚ
LazyTensorÚPackedTensorÚSymbolicDimÚShapeÚ
TensorTypeÚOptionalTypeÚSequenceTypeÚSparseTensorTypeÚTypeAndShapeÚValueÚAttrÚRefAttrÚNodeÚFunctionÚGraphÚ	GraphViewÚModelÚAttrFloat32ÚAttrFloat32sÚ	AttrGraphÚ
AttrGraphsÚ	AttrInt64Ú
AttrInt64sÚAttrSparseTensorÚAttrSparseTensorsÚ
AttrStringÚAttrStringsÚ
AttrTensorÚAttrTensorsÚAttrTypeProtoÚAttrTypeProtosÚInputÚArrayCompatibleÚDLPackCompatibleÚTensorProtocolÚValueProtocolÚModelProtocolÚNodeProtocolÚGraphProtocolÚGraphViewProtocolÚAttributeProtocolÚReferenceAttributeProtocolÚSparseTensorProtocolÚSymbolicDimProtocolÚShapeProtocolÚTypeProtocolÚMapTypeProtocolÚFunctionProtocolÚAttributeTypeÚDataTypeÚOperatorIdentifierÚTensorProtoTensorÚ
from_protoÚfrom_onnx_textÚto_protoÚto_onnx_textÚtensorÚnodeÚvalÚpassesÚloadÚsaveÚsave_safetensorsÚDEBUGÚset_value_magic_handleré    N)r   r   rE   r   r   r   r   )rC   rB   rD   )$r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r   r   r   r   r)   r
   r   r   r   r   r   r   r   r   r	   r   r   r   r   r   rJ   )r:   r;   )rF   rG   )r*   r2   r+   r9   r0   r1   r8   r.   r/   r<   r3   r6   r4   r5   r,   r7   r-   )rH   )r=   r?   r>   rA   r@   FÚreturnc                  C   s8   t ƒ } tD ]}| | }t|dƒrt|tjƒst|_qdS )zESet the module of all functions in this module to this public module.Ú
__module__N)ÚglobalsÚ__all__ÚhasattrÚ
isinstanceÚtypesÚGenericAliasÚ__name__rM   )Úglobal_dictÚnameÚobj© rX   úD/home/ubuntu/.local/lib/python3.10/site-packages/onnx_ir/__init__.pyÚ__set_module§   s   €ýrZ   z0.1.16)rL   N)WÚ__doc__rO   rR   Úonnx_irr   r   rE   r   r   r   r   Ú"onnx_ir._convenience._constructorsrC   rB   rD   Úonnx_ir._corer   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r   r   r   r   r)   r
   r   r   r   r   r   r   r   r   r	   r   r   r   r   r   rJ   Úonnx_ir._enumsr:   r;   Úonnx_ir._iorF   rG   Úonnx_ir._protocolsr*   r2   r+   r9   r0   r1   r8   r.   r/   r<   r3   r6   r4   r5   r,   r7   r-   Úonnx_ir._safetensorsrH   Úonnx_ir.serder=   r?   r>   rA   r@   rI   rZ   Ú__version__rX   rX   rX   rY   Ú<module>   s   Z$˜&L
	