
    pi9                        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             ZCyu)v    )	dataclassfield)IntEnum)	AnnotatedOptional)_Union_union_dataclass)         c                   X    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y)
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     \/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/torch/_export/serde/schema.pyr   r      sa    GDDE
CDDEFKLMDHFLJNNr6   r   c                   ,    e Zd ZdZdZdZdZdZdZdZ	dZ
y	)
Layoutr   r   r   r   r   r   r   r   N)r   r    r!   Unknown	SparseCoo	SparseCsr	SparseCsc	SparseBsr	SparseBsc_mkldnnStridedr5   r6   r7   r9   r9   (   s*    GIIIIIGGr6   r9   c                        e Zd ZdZdZdZdZdZy)MemoryFormatr   r   r   r   r   N)r   r    r!   r:   ContiguousFormatChannelsLastChannelsLast3dPreserveFormatr5   r6   r7   rC   rC   3   s    GLNNr6   rC   c                   @    e Zd ZU eedf   ed<   dZeee   df   ed<   y)Devicer   typeN   index)	r   r    r!   r   str__annotations__rL   r   intr5   r6   r7   rI   rI   ;   s)    
CG
*.E9Xc]B&'.r6   rI   c                   J    e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   y)SymExprHintr   as_intrK   as_boolr   as_floatN)r   r    r!   r   rO   rN   boolfloatr5   r6   r7   rQ   rQ   A   s/    c2gtRx  r	""r6   rQ   c                   @    e Zd ZU eedf   ed<   dZeee   df   ed<   y)SymExprr   expr_strNrK   hint)	r   r    r!   r   rM   rN   rZ   r   rQ   r5   r6   r7   rX   rX   L   s*    R  15D)H[)2-
.5r6   rX   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymIntr   as_exprrK   rR   N)r   r    r!   r   rX   rN   rO   r5   r6   r7   r\   r\   R   s     w{##c2gr6   r\   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymFloatr   r]   rK   rT   N)r   r    r!   r   rX   rN   rV   r5   r6   r7   r_   r_   X   s     w{##r	""r6   r_   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)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<   y)
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   su    Z^$$T&\2%&&T2X&&fbj!!tF|R'((fbj))fbj!!r6   rc   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymIntArgumentr   as_namerK   rR   N)r   r    r!   r   rM   rN   rO   r5   r6   r7   rq   rq   u   s     sBwc2gr6   rq   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymFloatArgumentr   rr   rK   rT   N)r   r    r!   r   rM   rN   rV   r5   r6   r7   rt   rt      s     sBwr	""r6   rt   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)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<   y)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<   y)TokenArgumentr   ry   Nrz   r5   r6   r7   r}   r}      r{   r6   r}   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)OptionalTensorArgumentrK   	as_tensorr   as_noneN)r   r    r!   r   rx   rN   rU   r5   r6   r7   r   r      s!    +,,tRx  r6   r   c                   2    e Zd ZU eedf   ed<   ed   ed<   y)GraphArgumentr   ry   )GraphrK   graphNrz   r5   r6   r7   r   r      s    
CG
[!!r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)CustomObjArgumentr   ry   rK   	class_fqnNrz   r5   r6   r7   r   r      s     
CG
b!!r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)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<   y7)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c2gtCy"}%%r	""eb))c""$s)S.)).#-..4/455j#o..c 122%%%%tSy!!T
C((?C/00D13677s*++"4(>#?#DEE.34438$$,c122T"23S899!"8#"=>>,+,,r6   r   c                       e Zd ZdZdZdZy)ArgumentKindr   r   r   N)r   r    r!   r"   
POSITIONALKEYWORDr5   r6   r7   r   r      s    GJGr6   r   c                   T    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<   y)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.
/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<   y	)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      sk    c2gd=)2-..tH~r)**S#X*++AE8D>2+=!>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<   y)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      s    d8nb())tH~r)**T$Z^$$T#z/2B677d3;/344tCL12566
 49YtRx08EJFyc+<&<!=r!AB  <AQU;ViS(] 3R 78Vr6   r   c                   "    e Zd ZU eedf   ed<   y)UserInputSpecr   r   Nr   r    r!   r   r   rN   r5   r6   r7   r   r     s     
8R<	  r6   r   c                   r    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
<   y)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   	  sM    tRx  c2gr	""b!!tRx  r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToConstantInputSpecr   ry   rK   valueN)r   r    r!   r   rM   rN   r   r5   r6   r7   r   r     s!    
CG
]B&''r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToParameterSpecr   r   rK   parameter_nameNr   r    r!   r   rx   rN   rM   r5   r6   r7   r   r     !    	>2%	&&c2g&&r6   r   c                   J    e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   y)InputToBufferSpecr   r   rK   buffer_namer   
persistentN)r   r    r!   r   rx   rN   rM   rU   r5   r6   r7   r   r     s0    	>2%	&&37##$(##r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToTensorConstantSpecr   r   rK   tensor_constant_nameNr   r5   r6   r7   r   r   %  s!    	>2%	&&#CG,,r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)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<   y)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<   y)	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  sw    -+,,-r122'+,,8"<==.233^R'((6:;;r6   r   c                   "    e Zd ZU eedf   ed<   y)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<   y)LossOutputSpecr   r   N)r   r    r!   r   rx   rN   r5   r6   r7   r   r   F  s    	>2%	&&r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)BufferMutationSpecr   r   rK   r   Nr   r5   r6   r7   r   r   K  s!    	>2%	&&37##r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)ParameterMutationSpecr   r   rK   r   Nr   r5   r6   r7   r   r   Q  r   r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)GradientToParameterSpecr   r   rK   r   Nr   r5   r6   r7   r  r  W  r   r6   r  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)GradientToUserInputSpecr   r   rK   user_input_nameNr   r5   r6   r7   r  r  ]  !    	>2%	&&sBw''r6   r  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)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<   y)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<   y)
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$%<b%@AA%&=r&ABB"#8"#<==_b())!"7";<<r6   r  c                   B    e Zd ZU eee   df   ed<   eee   df   ed<   y)GraphSignaturer   input_specsrK   output_specsN)r   r    r!   r   ro   r   rN   r  r5   r6   r7   r  r  z  s+    4	?B.//D,b011r6   r  c                   B    e Zd ZU eee   df   ed<   eee   df   ed<   y)RangeConstraintr   min_valrK   max_valN)r   r    r!   r   r   rO   rN   r5   r6   r7   r  r    s*    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<   y	)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    sk    d8nb())tH~r)** sBwR   =Ay$s)!4b!89@r6   r  c                   @    e Zd ZU eedf   ed<   dZeee   df   ed<   y)ModuleCallEntryr   fqnNr   	signature)	r   r    r!   r   rM   rN   r#  r   r  r5   r6   r7   r!  r!    s+    	37	>BIy"56:;Br6   r!  c                   (    e Zd ZU eee   df   ed<   y)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<   y)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!:;;.3D.IHiS#X*+IJOK	$sM/A*BB*F G r6   r(  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SchemaVersionr   majorrK   minorN)r   r    r!   r   rO   rN   r5   r6   r7   r,  r,    s,    R  R r6   r,  c                       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<   y)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 c?&:!;R!?@@mR/00*/*EIycB'E(/M9S"W%/,1$,GK49b=)Gr6   r0  c                   d    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<   y	)
PayloadMetar   	path_namerK   is_paramr   
use_picklerg   tensor_metaN)	r   r    r!   r   rM   rN   rU   r   rc   r5   r6   r7   r9  r9    sH     b!!b!! $(##8J/344r6   r9  c                   ,    e Zd ZU eeeef   df   ed<   y)PayloadConfigr   configN)r   r    r!   r   r   rM   r9  rN   r5   r6   r7   r?  r?    s    d3+,b0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<   y)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?C8C=!+;!<C ?Cy$):;Br6   rB  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)ExternKernelNoder   ry   rK   nodeN)r   r    r!   r   rM   rN   r   r5   r6   r7   rJ  rJ    s"     CG

D"H
r6   rJ  c                   (    e Zd ZU eee   df   ed<   y)ExternKernelNodesr   r   N)r   r    r!   r   ro   rJ  rN   r5   r6   r7   rM  rM    s    T*+R/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     su   )  & >  
 ,W 7  / / /
 #& # # 6 6 6
 V  
 #v # #
 !f ! !
 " " "  V   #v # # !f ! !
       !V ! !
 " " "
 " " "
    -v - -<7  7 7 7 F F F W W W$ ! ! !
 !F ! ! ( ( (
 ' ' '
 $ $ $ - - -
 ( ( (
 & & & < < < ! ! ! ' ' ' $ $ $
 ' ' '
 ' ' '
 ( ( (
 ( ( (
 & & & = = = 2 2 2
 * * *
 A A A C C C
 * * *        H H H$ 	5 	5 	5 2 2 2 C C C*    1 1 1r6   