
    `i9                     0	   d dl mZmZ d dlmZ d dlmZmZ d dlm	Z	m
Z
 dZdZ G d de          Z G d	 d
e          Z G d de          Ze G d d                      Ze
 G d de	                      Ze G d d                      Ze
 G d de	                      Ze
 G d de	                      Ze
 G d de	                      Ze G d d                      Ze
 G d de	                      Ze
 G d de	                      Ze
 G d d e	                      Ze G d! d"                      Ze G d# d$                      Ze
 G d% d&e	                      Ze G d' d(                      Ze G d) d*                      Ze G d+ d,                      Ze
 G d- d.e	                      Z  G d/ d0e          Z!e G d1 d2                      Z"e G d3 d4                      Z#e G d5 d6                      Z$e G d7 d8                      Z%e
 G d9 d:e	                      Z&e G d; d<                      Z'e G d= d>                      Z(e G d? d@                      Z)e G dA dB                      Z*e G dC dD                      Z+e G dE dF                      Z,e
 G dG dHe	                      Z-e G dI dJ                      Z.e G dK dL                      Z/e G dM dN                      Z0e G dO dP                      Z1e G dQ dR                      Z2e G dS dT                      Z3e G dU dV                      Z4e G dW dX                      Z5e
 G dY dZe	                      Z6e G d[ d\                      Z7e G d] d^                      Z8e G d_ d`                      Z9e G da db                      Z:e G dc dd                      Z;e G de df                      Z<e G dg dh                      Z=e G di dj                      Z>e G dk dl                      Z?e G dm dn                      Z@e G do dp                      ZAe G dq dr                      ZBe G ds dt                      ZCduS )v    )	dataclassfield)IntEnum)	AnnotatedOptional)_Union_union_dataclass)         c                   Z    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ZdZdZdZdZdS )
ScalarTyper   r                     r
   	   
                            N)__name__
__module____qualname__UNKNOWNBYTECHARSHORTINTLONGHALFFLOATDOUBLECOMPLEXHALFCOMPLEXFLOATCOMPLEXDOUBLEBOOLBFLOAT16UINT16FLOAT8E4M3FN
FLOAT8E5M2FLOAT8E4M3FNUZFLOAT8E5M2FNUZ     n/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/_export/serde/schema.pyr   r      sm        GDDE
CDDEFKLMDHFLJNNNNr6   r   c                   .    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!   Unknown	SparseCoo	SparseCsr	SparseCsc	SparseBsr	SparseBsc_mkldnnStridedr5   r6   r7   r9   r9   (   s6        GIIIIIGGGGr6   r9   c                   "    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:   ContiguousFormatChannelsLastChannelsLast3dPreserveFormatr5   r6   r7   rC   rC   3   s(        GLNNNNr6   rC   c                   T    e Zd ZU eedf         ed<   dZeee         df         ed<   dS )Devicer   typeN   index)	r   r    r!   r   str__annotations__rL   r   intr5   r6   r7   rI   rI   ;   sE         
CG
*.E9Xc]B&'.....r6   rI   c                   ^    e Zd ZU eedf         ed<   eedf         ed<   eedf         ed<   dS )SymExprHintr   as_intrK   as_boolr   as_floatN)r   r    r!   r   rO   rN   boolfloatr5   r6   r7   rQ   rQ   A   sQ         c2gtRx    r	""""""r6   rQ   c                   T    e Zd ZU eedf         ed<   dZeee         df         ed<   dS )SymExprr   expr_strNrK   hint)	r   r    r!   r   rM   rN   rZ   r   rQ   r5   r6   r7   rX   rX   L   sF         R    15D)H[)2-
.55555r6   rX   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )SymIntr   as_exprrK   rR   N)r   r    r!   r   rX   rN   rO   r5   r6   r7   r\   r\   R   s<         w{####c2gr6   r\   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )SymFloatr   r]   rK   rT   N)r   r    r!   r   rX   rN   rV   r5   r6   r7   r_   r_   X   s<         w{####r	""""""r6   r_   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )SymBoolr   r]   rK   rS   N)r   r    r!   r   rX   rN   rU   r5   r6   r7   ra   ra   ^   s<         w{####tRx      r6   ra   c                       e Zd ZU eedf         ed<   eee         df         ed<   eedf         ed<   ee	df         ed<   eee         d	f         ed
<   eedf         ed<   ee
df         ed<   dS )
TensorMetar   dtyperK   sizesr   requires_grad(   device2   strides<   storage_offsetF   layoutN)r   r    r!   r   r   rN   listr\   rU   rI   r9   r5   r6   r7   rc   rc   d   s         Z^$$$$T&\2%&&&&T2X&&&&fbj!!!!tF|R'((((fbj))))fbj!!!!!!r6   rc   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )SymIntArgumentr   as_namerK   rR   N)r   r    r!   r   rM   rN   rO   r5   r6   r7   rq   rq   u   s<         sBwc2gr6   rq   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )SymFloatArgumentr   rr   rK   rT   N)r   r    r!   r   rM   rN   rV   r5   r6   r7   rt   rt      s<         sBwr	""""""r6   rt   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )SymBoolArgumentr   rr   rK   rS   N)r   r    r!   r   rM   rN   rU   r5   r6   r7   rv   rv      s<         sBwtRx      r6   rv   c                   *    e Zd ZU eedf         ed<   dS )TensorArgumentr   nameNr   r    r!   r   rM   rN   r5   r6   r7   rx   rx      '         
CG
r6   rx   c                   *    e Zd ZU eedf         ed<   dS )TokenArgumentr   ry   Nrz   r5   r6   r7   r}   r}      r{   r6   r}   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )OptionalTensorArgumentrK   	as_tensorr   as_noneN)r   r    r!   r   rx   rN   rU   r5   r6   r7   r   r      s=         +,,,,tRx      r6   r   c                   @    e Zd ZU eedf         ed<   ed         ed<   dS )GraphArgumentr   ry   )GraphrK   graphNrz   r5   r6   r7   r   r      s8         
CG
[!!!!!!r6   r   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )CustomObjArgumentr   ry   rK   	class_fqnNrz   r5   r6   r7   r   r      s<         
CG
b!!!!!!r6   r   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )ComplexValuer   realrK   imagN)r   r    r!   r   rV   rN   r5   r6   r7   r   r      s<         
E2I

E2I
r6   r   c                   :   e Zd ZU eedf         ed<   eedf         ed<   eee         df         ed<   eedf         ed<   eee         d	f         ed
<   ee	df         ed<   eee	         df         ed<   ee
df         ed<   eee
         df         ed<   eedf         ed<   eee         df         ed<   eedf         ed<   eedf         ed<   eedf         ed<   eedf         ed<   eedf         ed <   eee         d!f         ed"<   eed#f         ed$<   eee         d%f         ed&<   eed'f         ed(<   eee         d)f         ed*<   eed+f         ed,<   ee
d-f         ed.<   eed/f         ed0<   eee         d1f         ed2<   eed3f         ed4<   eed5f         ed6<   d7S )8Argumentr   r   rK   r   r   
as_tensorsri   rR   rm   as_intsP   rT   Z   	as_floatsd   	as_stringe   
as_stringsn   
as_sym_intx   as_sym_ints   as_scalar_type   as_memory_format   	as_layout   	as_device   rS      as_bools   as_sym_bool   as_sym_bools   as_graph   as_optional_tensors   as_custom_obj   as_operator   as_sym_float   as_sym_floats   as_optional_tensori  
as_complexN)r   r    r!   r   rU   rN   rx   ro   rO   rV   rM   rq   r   rC   r9   rI   rv   r   r   r   rt   r   r5   r6   r7   r   r      s        tRx    +,,,,$~.23333c2gtCy"}%%%%r	""""eb))))c""""$s)S.)))).#-....4/45555j#o....c 12222%%%%%%%%tSy!!!!T
C((((?C/0000D1367777s*++++"4(>#?#DEEEE.3444438$$$$,c12222T"23S89999!"8#"=>>>>,+,,,,,,r6   r   c                       e Zd ZdZdZdZdS )ArgumentKindr   r   r   N)r   r    r!   r"   
POSITIONALKEYWORDr5   r6   r7   r   r      s        GJGGGr6   r   c                   n    e Zd ZU eedf         ed<   eedf         ed<   dZeee	         df         ed<   dS )NamedArgumentr   ry   rK   argNr   kind)
r   r    r!   r   rM   rN   r   r   r   r   r5   r6   r7   r   r      s]          CG
	8R<	    26D)H\*B.
/66666r6   r   c                       e Zd ZU eedf         ed<   eee         df         ed<   eee         df         ed<   ee	eef         df         ed<   d	Z
eee         d
f         ed<   d	S )Noder   targetrK   inputsr   outputsrg   metadataNri   is_hop_single_tensor_return)r   r    r!   r   rM   rN   ro   r   r   dictr   r   rU   r5   r6   r7   r   r      s         c2gd=)2-....tH~r)****S#X*++++AE8D>2+=!>EEEEEr6   r   c                      e Zd ZU eee         df         ed<   eee         df         ed<   eee         df         ed<   eee	e
f         df         ed<   eee	ef         d	f         ed
<   eee	ef         df         ed<   dZeedf         ed<    ee          Zeee	ef         df         ed<    ee          Zeee	ef         df         ed<   dS )r   r   r   rK   r   r   nodesrg   tensor_valuesri   sym_int_valuesrk   sym_bool_valuesFrm   is_single_tensor_returndefault_factoryr   custom_obj_valuesr   sym_float_valuesN)r   r    r!   r   ro   r   rN   r   r   rM   rc   r\   ra   r   rU   r   r   r   r   r_   r5   r6   r7   r   r      sP        d8nb())))tH~r)****T$Z^$$$$T#z/2B67777d3;/34444tCL1256666
 49YtRx0888EJUF F Fyc+<&<!=r!AB    <A5QU;V;V;ViS(] 3R 78VVVVVr6   r   c                   *    e Zd ZU eedf         ed<   dS )UserInputSpecr   r   Nr   r    r!   r   r   rN   r5   r6   r7   r   r     s)          
8R<	      r6   r   c                       e Zd ZU eedf         ed<   eedf         ed<   eedf         ed<   eedf         ed<   eed	f         ed
<   dS )ConstantValuer   r   rK   rR   r   rT   rg   r   ri   rS   N)	r   r    r!   r   rU   rN   rO   rV   rM   r5   r6   r7   r   r   	  s{         tRx    c2gr	""""b!!!!tRx      r6   r   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )InputToConstantInputSpecr   ry   rK   valueN)r   r    r!   r   rM   rN   r   r5   r6   r7   r   r     s=         
CG
]B&''''''r6   r   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )InputToParameterSpecr   r   rK   parameter_nameNr   r    r!   r   rx   rN   rM   r5   r6   r7   r   r     =         	>2%	&&&&c2g&&&&&&r6   r   c                   ^    e Zd ZU eedf         ed<   eedf         ed<   eedf         ed<   dS )InputToBufferSpecr   r   rK   buffer_namer   
persistentN)r   r    r!   r   rx   rN   rM   rU   r5   r6   r7   r   r     sR         	>2%	&&&&37####$(######r6   r   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )InputToTensorConstantSpecr   r   rK   tensor_constant_nameNr   r5   r6   r7   r   r   %  s=         	>2%	&&&&#CG,,,,,,r6   r   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )InputToCustomObjSpecr   r   rK   custom_obj_nameN)r   r    r!   r   r   rN   rM   r5   r6   r7   r   r   +  s>         	$b(	))))sBw''''''r6   r   c                   *    e Zd ZU eedf         ed<   dS )InputTokenSpecr   r   Nr   r    r!   r   r}   rN   r5   r6   r7   r   r   1  (         	="$	%%%%%%r6   r   c                       e Zd ZU eedf         ed<   eedf         ed<   eedf         ed<   eedf         ed<   ee	d	f         ed
<   ee
df         ed<   eedf         ed<   dS )	InputSpecr   
user_inputrK   	parameterr   bufferrg   tensor_constantri   
custom_objrm   tokenrk   constant_inputN)r   r    r!   r   r   rN   r   r   r   r   r   r   r5   r6   r7   r   r   6  s         -+,,,,-r12222'+,,,,8"<====.23333^R'((((6:;;;;;;r6   r   c                   *    e Zd ZU eedf         ed<   dS )UserOutputSpecr   r   Nr   r5   r6   r7   r   r   A  s'         	8R<	      r6   r   c                   *    e Zd ZU eedf         ed<   dS )LossOutputSpecr   r   N)r   r    r!   r   rx   rN   r5   r6   r7   r   r   F  s(         	>2%	&&&&&&r6   r   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )BufferMutationSpecr   r   rK   r   Nr   r5   r6   r7   r   r   K  s=         	>2%	&&&&37######r6   r   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )ParameterMutationSpecr   r   rK   r   Nr   r5   r6   r7   r   r   Q  r   r6   r   c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )GradientToParameterSpecr   r   rK   r   Nr   r5   r6   r7   r  r  W  r   r6   r  c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )GradientToUserInputSpecr   r   rK   user_input_nameNr   r5   r6   r7   r  r  ]  =         	>2%	&&&&sBw''''''r6   r  c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )UserInputMutationSpecr   r   rK   r  Nr   r5   r6   r7   r  r  c  r  r6   r  c                   *    e Zd ZU eedf         ed<   dS )OutputTokenSpecr   r   Nr   r5   r6   r7   r	  r	  i  r   r6   r	  c                       e Zd ZU eedf         ed<   eedf         ed<   eedf         ed<   eedf         ed<   ee	d	f         ed
<   ee
df         ed<   eedf         ed<   eedf         ed<   dS )
OutputSpecr   user_outputrK   loss_outputr   buffer_mutationrg   gradient_to_parameterri   gradient_to_user_inputrk   user_input_mutationrm   r   r   parameter_mutationN)r   r    r!   r   r   rN   r   r   r  r  r  r	  r   r5   r6   r7   r  r  n  s         >2-....>2-....1256666$%<b%@AAAA%&=r&ABBBB"#8"#<====_b())))!"7";<<<<<<r6   r  c                   \    e Zd ZU eee         df         ed<   eee         df         ed<   dS )GraphSignaturer   input_specsrK   output_specsN)r   r    r!   r   ro   r   rN   r  r5   r6   r7   r  r  z  sG         4	?B.////D,b0111111r6   r  c                   \    e Zd ZU eee         df         ed<   eee         df         ed<   dS )RangeConstraintr   min_valrK   max_valN)r   r    r!   r   r   rO   rN   r5   r6   r7   r  r    sF         x}b())))x}b())))))r6   r  c                       e Zd ZU eee         df         ed<   eee         df         ed<   eedf         ed<   eedf         ed<   d	Zee	ee                  d
f         ed<   d	S )ModuleCallSignaturer   r   rK   r   r   in_specrg   out_specNri   forward_arg_names)
r   r    r!   r   ro   r   rN   rM   r  r   r5   r6   r7   r  r    s         d8nb())))tH~r)**** sBwR     =Ay$s)!4b!89@@@@@r6   r  c                   T    e Zd ZU eedf         ed<   dZeee         df         ed<   dS )ModuleCallEntryr   fqnNr   	signature)	r   r    r!   r   rM   rN   r#  r   r  r5   r6   r7   r!  r!    sG         	37	>BIy"56:;BBBBBr6   r!  c                   6    e Zd ZU eee         df         ed<   dS )NamedTupleDefr   field_namesN)r   r    r!   r   ro   rM   rN   r5   r6   r7   r%  r%    s+         49b=))))))r6   r%  c                       e Zd ZU eedf         ed<   eedf         ed<   eee         df         ed<    e	e
          Zee
eef         df         ed	<    e	e
          Zee
eef         d
f         ed<   dS )GraphModuler   r   ri   r#  rk   module_call_graphr   rg   r   rm   treespec_namedtuple_fieldsN)r   r    r!   r   r   rN   r  ro   r!  r   r   r   rM   r*  r%  r5   r6   r7   r(  r(    s         UBY+,,,, !o!6!:;;;;.3eD.I.I.IHiS#X*+IIIJO%K K K	$sM/A*BB*F G     r6   r(  c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )SchemaVersionr   majorrK   minorN)r   r    r!   r   rO   rN   r5   r6   r7   r,  r,    sT         R    R     r6   r,  c                   2   e Zd ZU eedf         ed<   eeeef         df         ed<   eeee	f         df         ed<   ee
df         ed<    ee	          Zeee         d
f         ed<   dZeedf         ed<    ee	          Zeee         df         ed<   dS )ExportedProgramr   graph_modulerK   opset_versionr   range_constraintsrk   schema_versionr   rm   	verifiersz<=2.4r   torch_versionr   guards_codeN)r   r    r!   r   r(  rN   r   rM   rO   r  r,  r   ro   r5  r6  r7  r5   r6   r7   r0  r0    s         KO,,,,T#s(^R/0000 c?&:!;R!?@@@@mR/0000*/%*E*E*EIycB'EEE(/M9S"W%///,1E$,G,G,GK49b=)GGGGGr6   r0  c                       e Zd ZU eedf         ed<   eedf         ed<   eedf         ed<   eee         df         ed<   d	S )
PayloadMetar   	path_namerK   is_paramr   
use_picklerg   tensor_metaN)	r   r    r!   r   rM   rN   rU   r   rc   r5   r6   r7   r9  r9    sp          b!!!!b!!!! $(####8J/3444444r6   r9  c                   :    e Zd ZU eeeef         df         ed<   dS )PayloadConfigr   configN)r   r    r!   r   r   rM   r9  rN   r5   r6   r7   r?  r?    s2         d3+,b0111111r6   r?  c                       e Zd ZU eedf         ed<   eee         df         ed<   eee         df         ed<   dZeee	         df         ed	<   dZ
eee	         d
f         ed<   dZeee         df         ed<   dS )AOTInductorModelPickleDatar   library_basenamer   input_namesr   output_namesNr   floating_point_input_dtyper   floating_point_output_dtyper   aot_inductor_model_is_cpu)r   r    r!   r   rM   rN   ro   rF  r   rO   rG  rH  rU   r5   r6   r7   rB  rB    s           Q'''' 49a<(((( DIqL))))
 ?C	(3-*: ;BBB?C8C=!+;!<CCC ?Cy$):;BBBBBr6   rB  c                   D    e Zd ZU eedf         ed<   eedf         ed<   dS )ExternKernelNoder   ry   rK   nodeN)r   r    r!   r   rM   rN   r   r5   r6   r7   rJ  rJ    s>          CG

D"H
r6   rJ  c                   6    e Zd ZU eee         df         ed<   dS )ExternKernelNodesr   r   N)r   r    r!   r   ro   rJ  rN   r5   r6   r7   rM  rM    s.         T*+R/000000r6   rM  N)Ddataclassesr   r   enumr   typingr   r   torch._export.serde.unionr   r	   SCHEMA_VERSIONTREESPEC_VERSIONr   r9   rC   rI   rQ   rX   r\   r_   ra   rc   rq   rt   rv   rx   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,  r0  r9  r?  rB  rJ  rM  r5   r6   r7   <module>rT     s   ) ( ( ( ( ( ( (       & & & & & & & & > > > > > > > >  
       ,    W       7    / / / / / / / /
 # # # # #& # # # 6 6 6 6 6 6 6 6
     V   
 # # # # #v # # #
 ! ! ! ! !f ! ! !
 " " " " " " " "      V    # # # # #v # # # ! ! ! ! !f ! ! !
                 ! ! ! ! !V ! ! !
 " " " " " " " "
 " " " " " " " "
         - - - - -v - - -<    7    7 7 7 7 7 7 7 7 F F F F F F F F W W W W W W W W$ ! ! ! ! ! ! ! !
 ! ! ! ! !F ! ! ! ( ( ( ( ( ( ( (
 ' ' ' ' ' ' ' '
 $ $ $ $ $ $ $ $ - - - - - - - -
 ( ( ( ( ( ( ( (
 & & & & & & & & < < < < < < < < ! ! ! ! ! ! ! ! ' ' ' ' ' ' ' ' $ $ $ $ $ $ $ $
 ' ' ' ' ' ' ' '
 ' ' ' ' ' ' ' '
 ( ( ( ( ( ( ( (
 ( ( ( ( ( ( ( (
 & & & & & & & & = = = = = = = = 2 2 2 2 2 2 2 2
 * * * * * * * *
 A A A A A A A A C C C C C C C C
 * * * * * * * *                  H H H H H H H H$ 	5 	5 	5 	5 	5 	5 	5 	5 2 2 2 2 2 2 2 2 C C C C C C C C*         1 1 1 1 1 1 1 1 1 1r6   