U
    zh                      @   s  d dl mZmZ d dlmZ d dlmZmZmZm	Z	 d dl
mZ dZdZG dd deZG d	d
 d
eZG dd deZeG dd dZeddG dd deZeG dd dZeddG dd deZeddG dd deZeG dd dZeddG dd deZeddG dd deZeG dd  d ZeG d!d" d"ZeddG d#d$ d$eZeG d%d& d&ZeG d'd( d(ZeddG d)d* d*eZeG d+d, d,ZeG d-d. d.Z eG d/d0 d0Z!eG d1d2 d2Z"eddG d3d4 d4eZ#eG d5d6 d6Z$eG d7d8 d8Z%eG d9d: d:Z&eG d;d< d<Z'eG d=d> d>Z(eG d?d@ d@Z)eddG dAdB dBeZ*eG dCdD dDZ+eG dEdF dFZ,eG dGdH dHZ-eG dIdJ dJZ.eG dKdL dLZ/eG dMdN dNZ0eG dOdP dPZ1eddG dQdR dReZ2eG dSdT dTZ3eG dUdV dVZ4eG dWdX dXZ5eG dYdZ dZZ6eG d[d\ d\Z7eG d]d^ d^Z8eG d_d` d`Z9daS )b    )	dataclassfield)IntEnum)DictListOptionalTuple)_Union)         c                   @   sD   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdS )
ScalarTyper   r      r      r
            	   
            N)__name__
__module____qualname__UNKNOWNZBYTEZCHARZSHORTINTLONGZHALFFLOATDOUBLEZCOMPLEXHALFZCOMPLEXFLOATZCOMPLEXDOUBLEZBOOLZBFLOAT16 r    r    L/var/www/html/venv/lib/python3.8/site-packages/torch/_export/serde/schema.pyr      s   r   c                   @   s,   e Zd ZdZdZdZdZdZdZdZ	dZ
d	S )
Layoutr   r   r   r   r   r
   r   r   N)r   r   r   UnknownZ	SparseCooZ	SparseCsrZ	SparseCscZ	SparseBsrZ	SparseBscZ_mkldnnZStridedr    r    r    r!   r"       s   r"   c                   @   s    e Zd ZdZdZdZdZdZdS )MemoryFormatr   r   r   r   r   N)r   r   r   r#   ZContiguousFormatZChannelsLastZChannelsLast3dZPreserveFormatr    r    r    r!   r$   +   s
   r$   c                   @   s&   e Zd ZU eed< dZee ed< dS )DevicetypeNindex)r   r   r   str__annotations__r'   r   intr    r    r    r!   r%   3   s   
r%   F)reprc                   @   s&   e Zd ZU eed< eed< eed< dS )SymExprHintas_intas_floatas_boolN)r   r   r   r*   r)   floatboolr    r    r    r!   r,   9   s   
r,   c                   @   s&   e Zd ZU eed< dZee ed< dS )SymExprZexpr_strNhint)r   r   r   r(   r)   r3   r   r,   r    r    r    r!   r2   D   s   
r2   c                   @   s   e Zd ZU eed< eed< dS )SymIntas_exprr-   N)r   r   r   r2   r)   r*   r    r    r    r!   r4   J   s   
r4   c                   @   s   e Zd ZU eed< eed< dS )SymBoolr5   r/   N)r   r   r   r2   r)   r1   r    r    r    r!   r6   P   s   
r6   c                   @   sN   e Zd ZU eed< ee ed< eed< eed< ee ed< eed< e	ed< dS )	
TensorMetaZdtypesizesZrequires_gradZdevicestridesZstorage_offsetZlayoutN)
r   r   r   r   r)   r   r4   r1   r%   r"   r    r    r    r!   r7   V   s   
r7   c                   @   s   e Zd ZU eed< eed< dS )SymIntArgumentas_namer-   N)r   r   r   r(   r)   r*   r    r    r    r!   r:   g   s   
r:   c                   @   s   e Zd ZU eed< eed< dS )SymBoolArgumentr;   r/   N)r   r   r   r(   r)   r1   r    r    r    r!   r<   s   s   
r<   c                   @   s   e Zd ZU eed< dS )TensorArgumentnameNr   r   r   r(   r)   r    r    r    r!   r=   y   s   
r=   c                   @   s   e Zd ZU eed< dS )TokenArgumentr>   Nr?   r    r    r    r!   r@   ~   s   
r@   c                   @   s"   e Zd ZU eed< ed ed< dS )OptionalTensorArgument	as_tensorr    as_noneN)r   r   r   r=   r)   r   r    r    r    r!   rA      s   
rA   c                   @   s   e Zd ZU eed< ded< dS )GraphArgumentr>   GraphgraphNr?   r    r    r    r!   rD      s   
rD   c                   @   s   e Zd ZU eed< eed< dS )CustomObjArgumentr>   Z	class_fqnNr?   r    r    r    r!   rG      s   
rG   c                   @   s   e Zd ZU ed ed< eed< ee ed< eed< ee ed< eed< ee ed< e	ed	< ee	 ed
< e
ed< ee
 ed< eed< eed< eed< eed< eed< ee ed< eed< ee ed< eed< ee ed< eed< e	ed< dS )Argumentr    rC   rB   Z
as_tensorsr-   Zas_intsr.   Z	as_floats	as_stringZ
as_stringsZ
as_sym_intZas_sym_intsZas_scalar_typeZas_memory_formatZ	as_layoutZ	as_devicer/   Zas_boolsZas_sym_boolZas_sym_boolsZas_graphZas_optional_tensorsZas_custom_objZas_operatorN)r   r   r   r   r)   r=   r   r*   r0   r(   r:   r   r$   r"   r%   r1   r<   rD   rA   rG   r    r    r    r!   rH      s.   
rH   c                   @   s   e Zd ZU eed< eed< dS )NamedArgumentr>   argN)r   r   r   r(   r)   rH   r    r    r    r!   rJ      s   
rJ   c                   @   s>   e Zd ZU eed< ee ed< ee ed< eeef ed< dS )NodetargetinputsoutputsmetadataN)	r   r   r   r(   r)   r   rJ   rH   r   r    r    r    r!   rL      s   
rL   c                   @   s   e Zd ZU ee ed< ee ed< ee ed< eee	f ed< eee
f ed< eeef ed< dZeed< eed	Zeeef ed
< dS )rE   rN   rO   ZnodesZtensor_valuesZsym_int_valuesZsym_bool_valuesFis_single_tensor_return)default_factorycustom_obj_valuesN)r   r   r   r   rH   r)   rL   r   r(   r7   r4   r6   rQ   r1   r   dictrS   rG   r    r    r    r!   rE      s   
rE   c                   @   s   e Zd ZU eed< dS )UserInputSpecrK   Nr   r   r   rH   r)   r    r    r    r!   rU      s   
rU   c                   @   s:   e Zd ZU ed ed< eed< eed< eed< eed< dS )ConstantValuer    rC   r-   r.   rI   r/   N)	r   r   r   r   r)   r*   r0   r(   r1   r    r    r    r!   rW      s
   
rW   c                   @   s   e Zd ZU eed< eed< dS )ConstantInputSpecr>   valueN)r   r   r   r(   r)   rW   r    r    r    r!   rX      s   
rX   c                   @   s   e Zd ZU eed< eed< dS )InputToParameterSpecrK   parameter_nameNr   r   r   r=   r)   r(   r    r    r    r!   rZ      s   
rZ   c                   @   s&   e Zd ZU eed< eed< eed< dS )InputToBufferSpecrK   buffer_name
persistentN)r   r   r   r=   r)   r(   r1   r    r    r    r!   r]      s   
r]   c                   @   s   e Zd ZU eed< eed< dS )InputToTensorConstantSpecrK   Ztensor_constant_nameNr\   r    r    r    r!   r`      s   
r`   c                   @   s   e Zd ZU eed< eed< dS )InputToCustomObjSpecrK   Zcustom_obj_nameN)r   r   r   rG   r)   r(   r    r    r    r!   ra      s   
ra   c                   @   s   e Zd ZU eed< dS )InputTokenSpecrK   Nr   r   r   r@   r)   r    r    r    r!   rb     s   
rb   c                   @   sF   e Zd ZU eed< eed< eed< eed< eed< e	ed< e
ed< dS )		InputSpecZ
user_inputZ	parameterbufferZtensor_constantZ
custom_objtokenZconstant_inputN)r   r   r   rU   r)   rZ   r]   r`   ra   rb   rX   r    r    r    r!   rd     s   
rd   c                   @   s   e Zd ZU eed< dS )UserOutputSpecrK   NrV   r    r    r    r!   rg     s   
rg   c                   @   s   e Zd ZU eed< dS )LossOutputSpecrK   N)r   r   r   r=   r)   r    r    r    r!   rh     s   
rh   c                   @   s   e Zd ZU eed< eed< dS )BufferMutationSpecrK   r^   Nr\   r    r    r    r!   ri     s   
ri   c                   @   s   e Zd ZU eed< eed< dS )GradientToParameterSpecrK   r[   Nr\   r    r    r    r!   rj   #  s   
rj   c                   @   s   e Zd ZU eed< eed< dS )GradientToUserInputSpecrK   user_input_nameNr\   r    r    r    r!   rk   )  s   
rk   c                   @   s   e Zd ZU eed< eed< dS )UserInputMutationSpecrK   rl   Nr\   r    r    r    r!   rm   /  s   
rm   c                   @   s   e Zd ZU eed< dS )OutputTokenSpecrK   Nrc   r    r    r    r!   rn   5  s   
rn   c                   @   sF   e Zd ZU eed< eed< eed< eed< eed< e	ed< e
ed< dS )	
OutputSpecZuser_outputZloss_outputZbuffer_mutationZgradient_to_parameterZgradient_to_user_inputZuser_input_mutationrf   N)r   r   r   rg   r)   rh   ri   rj   rk   rm   rn   r    r    r    r!   ro   :  s   
ro   c                   @   s&   e Zd ZU ee ed< ee ed< dS )GraphSignatureZinput_specsZoutput_specsN)r   r   r   r   rd   r)   ro   r    r    r    r!   rp   E  s   
rp   c                   @   s   e Zd ZU eed< eed< dS )RangeConstraintZmin_valZmax_valNr   r   r   r*   r)   r    r    r    r!   rq   K  s   
rq   c                   @   s6   e Zd ZU ee ed< ee ed< eed< eed< dS )ModuleCallSignaturerN   rO   Zin_specZout_specN)r   r   r   r   rH   r)   r(   r    r    r    r!   rs   Q  s   
rs   c                   @   s&   e Zd ZU eed< dZee ed< dS )ModuleCallEntryZfqnN	signature)r   r   r   r(   r)   ru   r   rs   r    r    r    r!   rt   \  s   
rt   c                   @   s*   e Zd ZU eed< eed< ee ed< dS )GraphModulerF   ru   Zmodule_call_graphN)r   r   r   rE   r)   rp   r   rt   r    r    r    r!   rv   b  s   
rv   c                   @   s   e Zd ZU eed< eed< dS )SchemaVersionmajorminorNrr   r    r    r    r!   rw   n  s   
rw   c                   @   sF   e Zd ZU eed< eeef ed< eeef ed< e	ed< eed< dS )ExportedProgramZgraph_moduleZopset_versionZrange_constraintsZschema_versiondialectN)
r   r   r   rv   r)   r   r(   r*   rq   rw   r    r    r    r!   rz   t  s
   
rz   N):dataclassesr   r   enumr   typingr   r   r   r   Ztorch._export.serde.unionr	   ZSCHEMA_VERSIONZTREESPEC_VERSIONr   r"   r$   r%   r,   r2   r4   r6   r7   r:   r<   r=   r@   rA   rD   rG   rH   rJ   rL   rE   rU   rW   rX   rZ   r]   r`   ra   rb   rd   rg   rh   ri   rj   rk   rm   rn   ro   rp   rq   rs   rt   rv   rw   rz   r    r    r    r!   <module>   s   



