+
    &j              
         ^ RI Ht ^ RIHt ^ RIHt ^ RIHt ^ RIH	t	H
t
Ht ^ RIHt ^ RIHtHtHtHtHtHtHtHtHtHt ]'       d	   ^ RIHtHt RYR
 R llt]! RR7       ! R R4      4       t]! RR7       ! R R4      4       t]! RR7       ! R R]4      4       t]! RR7       ! R R4      4       t ]! RR7       ! R R] 4      4       t!]! RR7       ! R R4      4       t"]! RR7       ! R R4      4       t#]! RR7       ! R R4      4       t$]! RR7       ! R R4      4       t%]! RR7       ! R  R!4      4       t&R"R#/R$ R% llt'R& R' lt(R(R#R)R/R* R+ llt)R, R- lt*R. R/ lt+R"R#R0R#/R1 R2 llt,R"R#R0R#/R3 R4 llt-R5 R6 lt.R7 R8 lt/R9 R: lt0R; R< lt1R= R> lt2RZR? R@ llt30 R[mt4RA RB lt5RC RD lt6RER	/RF RG llt7R)R/RH RI llt8R)R/RJ RK llt9R)R/RL RM llt:RNRORPRQRRRSRTRURVRU/t;R)R/RW RX llt<R	# )\    )annotations)	dataclass)TYPE_CHECKING)cpp)BindingCppSignatureCppSignatureGroup)pythonify_default)
ArgumentBaseTyBaseTypeFunctionSchemaListTypeNativeFunctionOptionalTypeReturnTypeVariant)IterableSequenceNc               (    V ^8  d   QhRRRRRRRR/# )   namestr	argumentszIterable[str]return_type
str | Nonereturn )formats   "k/Users/jameslopez/projects/CWCArchive/cwc-podcast/.venv/lib/python3.14/site-packages/torchgen/api/python.py__annotate__r"      s,     ? ?
?'?;E??    c                   \        V\        \        34      '       g   \        V4      pVe   RV 2MRpRV  RRP                  V4       RV R2p\	        V4      ^P8:  g!   \	        V4      ^ 8X  g   \        V4      R8X  d   V# RV  R2.R V 4       ORV R2NpR	P                  V4      p\
        ;QJ d    R
 V 4       F  '       d   K   RM	  RM! R
 V 4       4      '       d   V# VP                  R4      R,           # )Nz ->  zdef (, ): ...c              3  .   "   T F  pR V R2x  K  	  R# 5i)z    ,Nr   ).0args   & r!   	<genexpr>,format_function_signature.<locals>.<genexpr>   s     	-9CDQ-9s   
c              3  >   "   T F  p\        V4      ^P8*  x  K  	  R# 5i)P   N)len)r,   lines   & r!   r.   r/      s     
-ut3t9?u   FTz ...z  # fmt: skip
    ...self)
isinstancelisttuplejoinr3   allremovesuffix)r   r   r   sigliness   &&&  r!   format_function_signaturer@      s     i$//)$	*5*AD&rKa		),-Q{m5
AC
3x2~Y1,i0@I0M
 tfA	-9	- K=E
 ))E
C
s
-u
-sss
-u
---
 F#&>>>r#   T)frozenc                  "    ] tR t^t$ R]R&   RtR# )PythonReturnstuple[Return, ...]returnsr   N__name__
__module____qualname____firstlineno____annotations____static_attributes__r   r#   r!   rC   rC      s    r#   rC   c                  p    ] tR t^t$ R]R&   R]R&   R]R&   R]R&   RR	R
R/R R lltRR	RR	/R R lltRtR# )PythonArgumentr   r   r   typer   defaultdefault_initmethodFsymintTc               $    V ^8  d   QhRRRRRR/# )r   rR   boolrS   r   r   r   )r    s   "r!   r"   PythonArgument.__annotate__   s!     ( (d (D (C (r#   c               	R   \        V P                  VR 7      P                  RR4      P                  RR4      pV P                  pVR8X  d   VR9   d   V'       g   RpV P                  e9   RRRRR	RR
R/P                  V P                  V P                  4      pV RV RV 2# V RV 2# )rS   zconst r%   z &r7   inputnullptrNone::std::nulloptstd::nullopt{} =)TensorNumber)argument_type_strrO   replacer   rP   get)r7   rR   rS   type_strr   rP   s   &$$   r!   argument_strPythonArgument.argument_str   s    dii7WXr"WT2 	 yy 6>h*>>vD <<#6 &f	
 c$,,-  Zqay11Zq''r#   
deprecatedc               $    V ^8  d   QhRRRRRR/# )r   rR   rU   ri   r   r   r   )r    s   "r!   r"   rV     s$     0) 0)0)370)	0)r#   c               	   \        V P                  4      pV P                  pVR 8X  d   VR8X  d   V'       g   V'       g   RpVR8X  d
   VR,          pVR8X  d   V'       g   V R2P                  RR4      pT;'       d(    \	        V \
        4      ;'       d    V P                  R8H  pV P                  Ee.   V'       Eg%   \	        V P                  \        4      '       d   V P                  P                  \        \        P                  4      8X  d   V P                  P                  R	4      '       dv   V P                  P                  R
4      '       dU   RRP                  \        \         P"                  V P                  ^R P%                  R4      4      4      ,           R,           pM2RRRRRRRRRRRR/P'                  V P                  V P                  4      pV RV RV 2# V RV 2# )r7   ra   rY   from_out | None | None | Noner[   {}r&   r'   r+   r(   rZ   r\   r]   r^   zc10::MemoryFormat::Contiguouscontiguous_formatzQScheme::PER_TENSOR_AFFINEper_tensor_affine:  = )argument_type_str_pyirO   r   rd   r8   PythonOutArgumentrP   r   elemr   r   int
startswithendswithr;   mapr   stripsplitre   )r7   rR   ri   rf   r   treat_as_no_defaultrP   s   &$$    r!   argument_str_pyiPythonArgument.argument_str_pyi  s    )3yy 6>h(26*D6>CKD 5="7+334DiPH  ' '4!23' '& 	 <<#,?,?499h//IINNhvzz&::LL++C00LL))#.. $))C		4<<"3E3K3KC3P$QRRUXX 
 v$f"F&35H02E #dllDLL1  V2hZs7)44V2hZ((r#   r   N)rG   rH   rI   rJ   rK   rg   r   rL   r   r#   r!   rN   rN      sG    
I
J (U (4 (40) %0):?0) 0)r#   rN   c                  8    ] tR tRt$ R]R&   ]R R l4       tRtR# )ry   iF  tuple[PythonArgument, ...]outputsc                    V ^8  d   QhRRRR/# )r   r   r   r   PythonOutArgument | Noner   )r    s   "r!   r"   PythonOutArgument.__annotate__P  s      C C8 C=U Cr#   c                	   V '       g   R # \        V 4      pV^8X  d3   \        V ^ ,          P                  V ^ ,          P                  RR V R7      # V^8  dz   \        ;QJ d    R V  4       F  '       g   K   RM	  RM! R V  4       4      '       d   \        RV  24      h\        R\        \        \        P                  4      V4      RR V R7      # \        R4      h)	Nr[   )r   rO   rP   rQ   r   c              3  ^   "   T F#  qP                   P                  4       '       * x  K%  	  R # 5iNrO   is_tensor_liker,   as   & r!   r.   1PythonOutArgument.from_outputs.<locals>.<genexpr>^  s!     @1vv,,...s   +-TFzUnsupported output type: rn   z!Unexpected PythonOutArgument size)r3   ry   r   rO   anyRuntimeErrorr   r   r   ra   AssertionError)r   sizes   & r!   from_outputsPythonOutArgument.from_outputsO  s    7|19$QZ__QZ__!  AXs@@sss@@@@"%>wi#HII$hv}}5t<!  ABBr#   r   N)rG   rH   rI   rJ   rK   staticmethodr   rL   r   r#   r!   ry   ry   F  s     ('C Cr#   ry   c                      ] tR tRt$ R]R&   R]R&   R]R&   R]R&   R	]R
&   R]R&   R]R&   ]R R l4       tRRRR/R R lltR R ltR R lt	RRRR/R R llt
RR/R R lltRR/R R  lltR!tR"# )#PythonSignatureik  r   r   r   
input_argsinput_kwargsr   output_argsrC   rE   tensor_options_argsrU   rR   c                   V ^8  d   QhRR/# r   r   rU   r   )r    s   "r!   r"   PythonSignature.__annotate__  s      D r#   c                	    R # )Fr   r6   s   &r!   ri   PythonSignature.deprecated  s    r#   skip_outputsFskip_tensor_optionsc               $    V ^8  d   QhRRRRRR/# )r   r   rU   r   r   z.tuple[PythonArgument | PythonOutArgument, ...]r   )r    s   "r!   r"   r     s$     
 
#
BF
	7
r#   c               	0   . pVP                  V P                  4       VP                  V P                  4       V P                  e$   V'       g   VP	                  V P                  4       V'       g   VP                  V P
                  4       \        V4      # r   )extendr   r   r   appendr   r:   )r7   r   r   results   &$$ r!   r   PythonSignature.arguments  sj     <>doo&d''('MM$**+"MM$223V}r#   c                   V ^8  d   QhRR/# r   r   r{   r   )r    s   "r!   r"   r     s     % % %r#   c                	4    \        V P                  4       4      # r   )r3   r   r6   s   &r!   arguments_countPythonSignature.arguments_count  s    4>>#$$r#   c                   V ^8  d   QhRR/# r   r   )r    s   "r!   r"   r     s     = =C =r#   c                	`    \        V P                  4      \        V P                  4      ,           # r   )r3   r   r   r6   s   &r!   
output_idxPythonSignature.output_idx  s!    4??#c$*;*;&<<<r#   rS   Tc               $    V ^8  d   QhRRRRRR/# r   r   rU   rS   r   r   r   )r    s   "r!   r"   r     s"     	; 	;T 	;4 	;SV 	;r#   c               	6   V P                  VR 7      pV Uu. uF  qDP                  V P                  VR7      NK!  	  pp\        V P                  4      p\        V4      V8  d   VP                  VR4       V P                   RRP                  V4       R2# u upi )r   )rR   rS   *r&   r'   r(   )r   rg   rR   r3   r   insertr   r;   )r7   r   rS   argsr   schema_formalspositional_argcs   &$$    r!   signature_strPythonSignature.signature_str  s    ~~<~8GK%
GK!NN$++fN=t 	 %
 doo.~0!!/37))Adii78::%
   %Bc                    V ^8  d   QhRRRR/# r   r   rU   r   r   r   )r    s   "r!   r"   r     s     Q Q Q# Qr#   c               	|   V P                  VR 7      pV Uu. uF  q3P                  V P                  R7      NK   	  pp\        V P                  4      p\        V4      V8  d   VP                  VR4       \        V 4      pV P                  '       d   VP                  ^ R4       \        V P                  WF4      # u upi )r   rR   r   r7   	r   r   rR   r3   r   r   returns_str_pyir@   r   r7   r   r   r   r   r   returns_strs   &$     r!   signature_str_pyi!PythonSignature.signature_str_pyi  s    ~~<~8<@%
<@qdkk2D 	 %
 doo.~0!!/37 &d+;;;!!!V,(NPP%
s   $B9c                    V ^8  d   QhRRRR/# r   r   rU   r   r   r   )r    s   "r!   r"   r     s     !Q !Q !Q !Qr#   c               	   V P                  VR 7      pV Uu. uF  q3P                  V P                  R7      NK   	  ppV P                  4       pV^ 8X  d   R# \	        V P
                  4      pV^ ,          P                  p\        V\        4      '       d"   \        VP                  4      R9   d   V^8X  g   R# \        V\        4      '       g   \        R\        V4       24      hRV^ ,          P                  ,           R,           \        VP                  4      ,           V^ &   \        V 4      pV P                  '       d   VP                  ^ R4       \!        V P                  WH4      # u upi )r   r   NzExpected ListType, got r   ru   r7   )r{   SymInt)r   r   rR   r   r3   r   rO   r8   r   r   rz   r   r   rx   r   r   r@   )	r7   r   r   r   r   num_argsnum_positionalargsvararg_typer   s	   &$       r!   signature_str_pyi_vararg(PythonSignature.signature_str_pyi_vararg  s3   ~~<~8<@%
<@qdkk2D 	 %
 '')q= 11gll{H--K$$%)::"a' +x00 #:4;L:M!NOO$q',,%(=k>N>N(OO 	q &d+;;;!!!V,(NPP=%
s   $E'r   N)rG   rH   rI   rJ   rK   propertyri   r   r   r   r   r   r   rL   r   r#   r!   r   r   k  s     I +* -,))  43 L 
&+
IN
%=	;E 	;$ 	;Q Q !Qu !Q !Qr#   r   c                      ] tR tRt$ R]R&   R]R&   ]R R l4       tRR	R
R/R R lltRR	/R R lltRR	/R R llt	Rt
R# )PythonSignatureDeprecatedi  r   deprecated_schematuple[str, ...]deprecated_args_exprsc                   V ^8  d   QhRR/# r   r   )r    s   "r!   r"   &PythonSignatureDeprecated.__annotate__  s      D r#   c                	    R # Tr   r6   s   &r!   ri   $PythonSignatureDeprecated.deprecated      r#   r   FrS   Tc               $    V ^8  d   QhRRRRRR/# r   r   )r    s   "r!   r"   r     s"     
 
T 
4 
SV 
r#   c               	>    \         P                  WVR 7      R,           # ))r   rS   z|deprecated)r   r   )r7   r   rS   s   &$$r!   r   'PythonSignatureDeprecated.signature_str  s)    )) *  	
r#   c                    V ^8  d   QhRRRR/# r   r   )r    s   "r!   r"   r     s     
Q 
Q 
Q# 
Qr#   c               	6   V P                  VR 7      pV Uu. uF  q3P                  V P                  RR7      NK!  	  pp\        V P                  4      p\        V4      V8  d   VP                  VR4       \        V 4      p\        V P                  WF4      # u upi )r   T)rR   ri   r   r   r   s   &$     r!   r   +PythonSignatureDeprecated.signature_str_pyi  s    ~~<~8MQ%
MQdkkdCT 	 %
 doo.~0!!/37%d+(NPP%
r   c                    V ^8  d   QhRRRR/# r   r   )r    s   "r!   r"   r     s        r#   c               	    R # r   r   )r7   r   s   &$r!   r   2PythonSignatureDeprecated.signature_str_pyi_vararg  r   r#   r   N)rG   rH   rI   rJ   rK   r   ri   r   r   r   rL   r   r#   r!   r   r     sV     &% +* 
E 
$ 

Q 
Qu  r#   r   c                  ,    ] tR tRt$ R]R&   R]R&   RtR# )!PythonSignatureNativeFunctionPairi  r   	signaturer   functionr   NrF   r   r#   r!   r   r     s    r#   r   c                  L    ] tR tRt$ R]R&   R]R&   R]R&   ]R R	 l4       tR
tR# )PythonSignatureGroupi%  r   r   r   basezNativeFunction | Noneoutplacec               $    V ^8  d   QhRRRRRR/# )r   
functionalr   rn   z(PythonSignatureNativeFunctionPair | Noner   r   r   )r    s   "r!   r"   !PythonSignatureGroup.__annotate__3  s(     
 
5
 6
 
	
r#   c                	>   Vf#   \        VP                  VP                  R R7      # VP                  P                  P	                  4       pVP                  P
                  VR&   \        \        VP                  4      ! R/ VB VP                  VP                  R7      # )N)r   r   r   r   r   )r   r   r   __dict__copyr   rO   )clsr   rn   signature_kwargss   &&& r!   
from_pairsPythonSignatureGroup.from_pairs2  s     ;'$..((  ==11668
   44 	./ $3==)=,<=$$\\
 	
r#   r   N)rG   rH   rI   rJ   rK   classmethodr   rL   r   r#   r!   r   r   %  s,    
   $#
 
r#   r   c                  6    ] tR tRt$ R]R&   R]R&   R]R&   RtR# )	DispatchLambdaArgumentiU  r   r   rf   rU   
is_out_argr   NrF   r   r#   r!   r   r   U  s    
IMr#   r   c                  V    ] tR tRt$ R]R&   R]R&   R]R&   R]R&   ]R	 R
 l4       tRtR# )PythonArgParserOutputExpria  r   r   exprr{   indexrN   argumentc                   V ^8  d   QhRR/# )r   r   r   r   )r    s   "r!   r"   &PythonArgParserOutputExpr.__annotate__q  s     * *c *r#   c                	"    R V P                    R2# )z
_r.isNone(r(   )r  r6   s   &r!   is_none_expr&PythonArgParserOutputExpr.is_none_exprp  s    DJJ<q))r#   r   N)rG   rH   rI   rJ   rK   r   r  rL   r   r#   r!   r   r   a  s1     I I J * *r#   r   c                  ,    ] tR tRt$ R]R&   R]R&   RtR# )DispatchLambdaArgumentExprsiz  zSequence[str]exprsinitsr   NrF   r   r#   r!   r
  r
  z  s      r#   r
  rR   Fc               $    V ^8  d   QhRRRRRR/# )r   fr   rR   rU   r   r   r   )r    s   "r!   r"   r"     s&     N Nn N N, Nr#   c               D    \         P                  ! WR 7      P                  # )r   )r	   from_native_functionr   )r  rR   s   &$r!   _cpp_signaturer    s    11!CMMMr#   c                    V ^8  d   QhRRRR/# )r   r  r   r   rU   r   )r    s   "r!   r"   r"     s     7 7. 7T 7r#   c                F    V P                   P                  P                  R J# r   )funcr   tensor_options)r  s   &r!   has_tensor_optionsr    s    66**$66r#   simple_typerS   c               (    V ^8  d   QhRRRRRRRR/# )r   tr   r  rU   rS   r   r   r   )r    s   "r!   r"   r"     s,     =7 =7=7!=737=7=7r#   c                  \        V \        4      '       Ed_   V P                  \        P                  8X  d   R # V P                  \        P
                  8X  d   R# V P                  \        P                  8X  d   R# V P                  \        P                  \        P                  \        P                  \        P                  \        P                  \        P                  \        P                  \        P                  \        P                  \        P                   \        P"                  \        P$                  \        P&                  \        P(                  39   d   V P                  P                  # EM\        V \*        4      '       d   \-        V P.                  WR7      pV R2# \        V \0        4      '       Edo   V'       g   V P2                  MRp\        V P.                  4      R8X  d*   V P2                  f   \5        R4      hRV P2                   R	2# \        V P.                  4      R
8X  d   Ve   RV R2# R# \        V P.                  4      R8X  d!   V'       d   Ve   RV R2# R# Ve   RV R2# R# \        V P.                  4      R8X  d   Ve   RV R2# R# \        V P.                  4      R8X  d   Ve   RV R2# R# \        V P.                  4      R8X  d   V'       d   R# R# \        V P.                  4      R8X  d   Ve   RV R2# R# \-        V P.                  WR7      pRV R	2# \7        R\9        V 4       24      h)int64_tdoublezc10::string_view)r  rS   ?NrU   zbool ListType must have a sizez::std::array<bool,>r{   zIntArrayRef[]IntArrayRefr   zSymIntArrayRef[SymIntArrayRefra   zTensorList[
TensorListScalarzScalarList[
ScalarListTensor?z"c10::List<::std::optional<Tensor>>z*const c10::List<::std::optional<Tensor>> &DimnamezDimnameList[DimnameListz	ArrayRef<unrecognized type )r8   r   r   r   r{   floatr   ra   rU   QSchemer#  
ScalarType	GeneratorStorageLayoutDeviceDeviceIndexMemoryFormatr&  Streamr   r   rc   rz   r   r   r   r   repr)r  r  rS   rz   r   s   &$$  r!   rc   rc     s    !X66VZZVVv||#VVvzz!%VVMMKKNNMMNNMMMMNNMMMM
 
" 66;;#
& 
A|	$	$ [Pqz	Ax	 	 (qvvdqvv;& vv~$%EFF'xq11[E!-1-=\$q)P=P[H$151AodV1-GW 261AdV1-T}T[H$,0,<[a(N,N[H$,0,<[a(N,N[I%;C[I%-1-=\$q)P=P [P4&""
+DG95
66r#   c                    V ^8  d   QhRRRR/# )r   r  r   r   z
int | Noner   )r    s   "r!   r"   r"     s      $ : r#   c                z    V P                  4       pVe'   \        VP                  4      R8w  d   VP                  # R # )NrU   )is_list_liker   rz   r   )r  ls   & r!   argument_type_sizer8    s.    	A}QVV.vvr#   c                    V ^8  d   QhRRRR/# )r   r   r   r   rN   r   )r    s   "r!   r"   r"     s       ^ r#   c                    \        V P                  V P                  V P                  eF   \	        \        \        P                  ! V P                  V P                  RR7      4      4      R R7      # R R R7      # )NFrX   r   rO   rP   rQ   )rN   r   rO   rP   r   r
   r   default_expr)r   s   &r!   r  r    sg    VVVV yy$ !#"2"2199affU"STU 
 
 
 
r#   pyic               (    V ^8  d   QhRRRRRRRR/# )r   r  r   rR   rU   r=  r   r   r   )r    s   "r!   r"   r"     s,      "&59r#   c               F    \        V P                  V P                  WR 7      # ))category_overriderR   r=  )signature_from_schemar  r@  )r  rR   r=  s   &$$r!   r   r     s"     !	!"5"5f r#   c          
     ,    V ^8  d   QhRRRRRRRRRR	/# )
r   r  r   r@  r   rR   rU   r=  r   r   r   )r    s   "r!   r"   r"     sC     F F
F "F 	F
 
F Fr#   c          
       a aaa . pVP                  S P                  P                  4       V'       gH   S P                  P                  e0   VP	                  S P                  P                  P
                  4       VP                  S P                  P                  4       VP                  S P                  P                  4       VP                  S P                  P                  4       VP                  S P                  P                  4       S P                  P                   Uu0 uF  qUP                  kK  	  upoS P                  P                   Uu0 uF  qUP                  kK  	  upoS P                  P                   Uu0 uF  qUP                  kK  	  upo\        \        \
        \        V3R lV4      4      4      p\        \        \
        \        V3R lV4      4      4      p\        \        \
        \        V3R lV4      4      4      p\         ;QJ d0    R S P                  P"                   4       F  '       g   K   RM%	  RM!! R S P                  P"                   4       4      p	\         ;QJ d*    R S P%                  4        4       F  '       g   K   RM	  RM! R S P%                  4        4       4      '       d   \'        R4      h\         ;QJ d&    R	 S P(                   4       F  '       g   K   RM	  RM! R	 S P(                   4       4      p
\*        P                  ! S 4      pVR
8H  ;'       g    T
;'       d    V	'       * pVR9   ;'       g+    VP-                  R4      ;'       g    VP/                  R4      pVR8H  p. pV'       g
   V'       Ed   V'       Egz   R V 3R llpTP	                  \1        R\3        \5        \6        P8                  4      4      RV'       d   R MV! R4      R7      4       TP	                  \1        R\3        \5        \6        P:                  4      4      RV'       d   R MV! R4      R7      4       TP	                  \1        R\3        \5        \6        P<                  4      4      RV'       d   R MV! R4      ;'       g    RR7      4       VP	                  \1        R\3        \5        \6        P>                  4      4      RR R7      4       VP	                  \1        R\3        \5        \6        P>                  4      4      RR R7      4       \A        S P(                  R7      p\C        \E        S P                  P                  4      VV\F        PI                  V4      \        V4      VVR7      # u upi u upi u upi )Nc                "   < V P                   S9   # r   r   )r   input_arg_sets   &r!   <lambda>'signature_from_schema.<locals>.<lambda>  s    aff6Mr#   c                "   < V P                   S9   # r   rE  )r   kwarg_only_sets   &r!   rG  rH    s    qvv'?r#   c                "   < V P                   S9   # r   rE  )r   out_arg_sets   &r!   rG  rH     s    166[3Hr#   c              3  T   "   T F  qP                   P                  4       x  K   	  R # 5ir   r   r   s   & r!   r.   (signature_from_schema.<locals>.<genexpr>*  s"      )DA)D   &(TFc              3  >   "   T F  qP                   R 8H  x  K  	  R# 5i)requires_gradNrE  r   s   & r!   r.   rN  -  s     
L.K66_$.Kr5   zTargument named requires_grad is reserved, should not explicitly add it in the schemac              3  T   "   T F  qP                   P                  4       x  K   	  R # 5ir   r   r,   rs   & r!   r.   rN  4  s     J\FF1133\rO  factorynew__likedummyc                    V ^8  d   QhRRRR/# )r   r   r   r   r   r   )r    s   "r!   r"   +signature_from_schema.<locals>.__annotate__D  s     	E 	EC 	EJ 	Er#   c                   < SP                   P                  pVf   R # \        W4      pVP                  e   VP                  R8X  d   R # \        P
                  ! VP                  VP                  RR7      # )Nr[   FrX   )r   r  getattrrP   r   r<  rO   )r   	topt_argsr   r  s   &  r!   topt_default_init0signature_from_schema.<locals>.topt_default_initD  sZ    55I 	(Ayy AII$7##AIIqvveDDr#   dtyper[   r;  layoutdevicez$torch::tensors::get_default_device()
pin_memoryFalserQ  )rE   )r   r   r   r   r   rE   rR   )newlike)%r   r   pre_self_positionalself_argr   r  post_self_positionalpre_tensor_options_kwarg_onlypost_tensor_options_kwarg_onlyrn   flat_positionalr   flat_kwarg_onlyr:   r~   filterr   flat_non_outschema_order_arguments
ValueErrorrE   r   r|   r}   rN   r   r   r   r+  r.  r/  rU   rC   r   r   ry   r   )r  r@  rR   r=  r   r   r   r   r   has_tensor_input_arghas_tensor_returnr   is_factory_functionis_like_or_new_functionis_dummy_functionr   r^  rE   rF  rJ  rL  s   f$$$              @@@r!   rA  rA    s:    DKK223dnn--9DNN++445KK334KK<<= 	KK==>KK""#%)^^%C%CD%CVV%CDM&*nn&D&DE&Dff&DEN#'>>#5#56#5a66#56Ks8V,Mt%TUVJHf?FGL C&)H$"OPQG 3 )-)D)D333 )-)D)D  s
Ld.I.I.K
Lsss
Ld.I.I.K
LLLb
 	
 JT\\JJT\\JJD+y8  66"66  	_, 	" 	"??6"	" 	"==! 
 *W40266@Q@Q	E 	E 	""!(6+<+<"=>3D9J79S		
 	""!(6=="9:3D9J89T		
 	""!(6=="9: /  *(3 B BA	
 	""!!(6;;"78!		
 	""$!(6;;"78!		
 DLL1G !%227;!"56 W EE6s   'U<VVc                    V ^8  d   QhRRRR/# )r   rE   rD   r   z	list[str]r   )r    s   "r!   r"   r"     s     . ."4 . .r#   c                x   \        V 4      ^8:  g;   \        ;QJ d    R V  4       F  '       d   K   RM	  RM! R V  4       4      '       d   . # \        ;QJ d    R V  4       F  '       g   K   RM	  RM! R V  4       4      '       d   \        R4      hV  Uu. uF  p\	        VP
                  4      NK  	  up# u upi )   c              3  <   "   T F  qP                   R J x  K  	  R # 5ir   rE  rS  s   & r!   r.   'structseq_fieldnames.<locals>.<genexpr>  s     @1$   FTc              3  <   "   T F  qP                   R J x  K  	  R # 5ir   rE  rS  s   & r!   r.   r{    s     /w!vv~wr|  z)Unnamed field is not supported by codegen)r3   r<   r   rq  r   r   )rE   rT  s   & r!   structseq_fieldnamesr~    s    
7|qCC@@CCC@@@@	3/w/333/w/// HII%,-WAFFW---s   B7c                    V ^8  d   QhRRRR/# r   r  r   r   r   r   )r    s   "r!   r"   r"     s     E ET Ec Er#   c                   R p\        V \        4      '       d   V P                  p RpRp\        V \        4      '       Ed,   V P                  \
        P                  \
        P                  39   d   RpV P                  \
        P                  8X  d   RpEMV P                  \
        P                  8X  d   RpEMV P                  \
        P                  8X  d   RpEMV P                  \
        P                  8X  d   RpEMV P                  \
        P                  8X  d   RpEMV P                  \
        P                  8X  d   R	pEM`V P                  \
        P                  8X  d   R
pEM=V P                  \
        P                  8X  d   RpEMV P                  \
        P                   8X  d   RpEMV P                  \
        P"                  8X  d   RpEMV P                  \
        P$                  8X  d   RpEMV P                  \
        P&                  8X  d   RpEMV P                  \
        P(                  \
        P*                  \
        P,                  39   d   V P                  P                  pEM8\        V \.        4      '       Ed   \        V P                  4      R8X  d   V P0                  e   RMRpMV P3                  4       '       dI   \        V P                  \        4      '       d   RpRpMRpV P0                  e
   RV RV R2MRV RV R2pM\        V P                  4      R8X  d   RpMy\        V P                  4      R8X  d-   V P0                  e   \5        V P                  4      pV RV R2pM3\5        V P                  4      pRV R2pM\7        R\9        V 4       24      hV'       d   V R2P;                  R R4      pV# )!FTr%   _intz_int | SymInt_floatr   zNumber | _complex_dtype_bool_qscheme_layoutzDeviceLikeType | Nonememory_formatzstr | EllipsisType | NonezStorage | UntypedStorager{   z_int | _size_sizezTensor | Nonera   zTensor | tuple[z, ...] | list[r  tuple[r)  zSequence[_float]r   z | Sequence[z	Sequence[r(  ro   rp   )r8   r   rz   r   r   r   r{   r0  r   r)  r   r#  r+  rU   r*  r.  r/  r1  r&  r-  ra   r,  r2  r   r   r   rx   r   r3  rd   )r  add_optionalretelem_strrz   s   &    r!   rx   rx     s   L!\""FF
C!X66fjj&"4"455C66V]]"!CVVv||#CVVvzz!CVVv}}$%CVVv(((CVVv{{"CVVv~~%CVVv}}$CVVv}}$)CVVv***!CVVv~~%-CVVv~~%,CVVv'7'7GG&&++C	Ax	 	 qvv;%$%FF$6.GC !&&,//#*# 66% "(>(1EhZ~hZqA 
 [G#$C[H$);(0DF,tfA.C(0DdV1%C /Qy9::Wo%%&6	BJr#   c                    V ^8  d   QhRRRR/# r  r   )r    s   "r!   r"   r"     s     $ $4 $C $r#   c                   \        V \        4      '       d+   \        V P                  4      pV R 2P	                  RR 4      # \        V \
        4      '       dN   V P                  \        P                  8X  d   R# V P                  \        P                  8X  d   R# \        V 4      # \        V \        4      '       d   \        V P                  4      pRV R2# \        V 4      # )ro   rp   _devicer   r  z, ...])r8   r   return_type_str_pyirz   rd   r   r   r   r/  r&  rx   r   )r  inners   & r!   r  r    s     !\""#AFF+ (()99EE!X66V]]"VVv~~%(++!X#AFF+wf%% ##r#   c                    V ^8  d   QhRRRR/# )r   r   r   r   ztuple[str, str] | Noner   )r    s   "r!   r"   r"     s     4 4_ 49O 4r#   c                   V P                   P                    Uu. uF  p\        VP                  4      NK  	  ppV P                  p\	        V P                   P                   4      pV'       d   R RP                  V4       R2pRV RV R2.p\        WB4       F  w  rxVP                  RRV RV R	2.4       K!  	  VP                  R
RRV R2RRR\        V4       2R\        V4       2RRR.
4       RP                  V4      p	W93# R# u upi )r  r'   r  zclass r&   z):  # fmt: skipz    @propertyz    def z
(self) -> r)   z    def __new__(z        cls,z        sequence: r+   z    ) -> Self:  # fmt: skipz        ...z    n_fields: Final[_int] = z%    n_sequence_fields: Final[_int] = z%    n_unnamed_fields: Final[_int] = 0zG    def __init_subclass__(cls) -> NoReturn: ...  # prohibit subclassingr%   r0   N)	rE   r  rO   r   r~  r;   zipr   r3   )
r   rT  python_returnsstructseq_namefield_namesseq_typestructseq_def_linesr   ret_typestructseq_defs
   &         r!   returns_structseq_pyir    s<   ;D;L;L;T;TU;Ta)!&&1;TNU^^N&y'8'8'@'@AK DIIn56a8^$AhZ?
 "+>ND&&#tfJxj> ? 	"""$XJa0-.s;/?.@A7K8H7IJ7Y	
 		"56$ ,,g Vs   Dc                    V ^8  d   QhRRRR/# )r   r   r   r   r   r   )r    s   "r!   r"   r"   >  s     
 
 
3 
r#   c                |   \        V P                  P                  4      pV'       d   R V P                   2# V P                  P                   Uu. uF  p\        VP                  4      NK  	  pp\        V4      ^8  d    RRP                  V4      ,           R,           # \        V4      ^8X  d
   V^ ,          # R# u upi )ztorch.return_types.r  r'   r  r[   )r~  rE   r   r  rO   r3   r;   )r   r  rT  r  s   &   r!   r   r   >  s    &y'8'8'@'@AK$Y^^$455;D;L;L;T;TU;Ta)!&&1;TNU
>Q$))N33c99
>aa   Vs   B9c               (    V ^8  d   QhRRRRRRRR/# )	r   psr   r  r   rS   rU   r   z"tuple[DispatchLambdaArgument, ...]r   )r    s   "r!   r"   r"   s  s,     )5 )5)5*)548)5')5r#   c                f  a a \        S \        4      '       d   S P                  pMVP                  p\        P
                  ! VP
                  R VR VP                  R7      pVP
                  P                   Uu0 uF  qUP                  kK  	  upoR VV 3R llp\        \        Wd4      4      # u upi )F)r   faithfulrS   rR   cpp_no_default_argsc                    V ^8  d   QhRRRR/# )r   cpp_argr   r   r   r   )r    s   "r!   r"   *dispatch_lambda_args.<locals>.__annotate__  s     
 
W 
1G 
r#   c                &  < V P                   pV P                  S9   pSP                  '       d   V P                  R 8X  d   RpM8\        S4      ^8*  ;'       g    V'       * pV'       d   RR/P	                  W4      p\        V P                  VVR7      # )r7   zconst at::Tensor &zat::Tensor &
at::Tensor)r   rf   r   )rO   r   rR   r3   re   r   )r  rf   r   ensure_temp_safeout_argsr  s   &   r!   dispatch_lambda_arg1dispatch_lambda_args.<locals>.dispatch_lambda_arg  s    <<\\X-
999/+H  #8}1CC^"L#h)  &!
 	
r#   )r8   r   r   r  r   r   r  rn   r   r:   r~   )r  r  rS   schemacpp_argsr   r  r  s   f&&    @r!   dispatch_lambda_argsr  s  s     "/00%% }}""11H +1*:*:*>*>?*>Q&&*>?H
 
, (3443 @s   9B.c                    V ^8  d   QhRRRR/# r   r  r   r   r   r   )r    s   "r!   r"   r"     s      . S r#   c                X   \         ;QJ d(    . R  V P                  P                   4       F  NK  	  5M!! R  V P                  P                   4       4      p\        P                  ! VRR7      P                  4       pV\        9  d%   \        V P                  P                   RV 24      hV# )c              3  d   "   T F&  p\        VP                  VP                  R 4      x  K(  	  R # 5ir   )r   r   rO   rS  s   & r!   r.   -dispatch_lambda_return_str.<locals>.<genexpr>  s'      '.<qvvqvvt$$ns   .0TrX   z returns unsupported type )	r:   r  rE   r   returns_typecpp_typeSUPPORTED_RETURN_TYPESr   r   )r  returns_without_annotation
return_strs   &  r!   dispatch_lambda_return_strr    s    ( "' './ffnn' './ffnn' " !!"<TJSSUJ//affkk]*DZLQRRr#   c                    V ^8  d   QhRRRR/# r  r   )r    s   "r!   r"   r"     s     U U> Uc Ur#   c                   V P                   P                  4       p\        P                  ! V P                   VR 7      p\        P
                  V P                  9   d   RV 2# \        P                  V P                  9   d]   \        V 4      '       g@   V P                   P                  P                  P                  P                  R4      '       d   RpMRpV RV 2# \        RV P                    24      h))symint_overloadzself.rW  torchatz::z1could not dispatch, neither function nor method: )r  
has_symintr   r   r   rR   variantsr   r  r   r}   r   )r  rS   r   	namespaces   &   r!   cpp_dispatch_targetr    s    VV F88AFFF3D~~#tf~1::%a  AFFKK$4$4$9$9$B$B7$K$KIIBtf%%
J166(S
TTr#   python_signaturec               $    V ^8  d   QhRRRRRR/# )r   r  r   r  zPythonSignature | Noner   r   r   )r    s   "r!   r"   r"     s(       - 	r#   c                 a  \        S R R7      P                  4       pRp\        V\        4      '       g.   \        ;QJ d    . R V 4       F  NK  	  5M! R V 4       4      pM#\	        \        V 3R lVP                  4      4      p\        P                  S P                  9   d    \	        \        RP                  V4      4      pV# )Fr   c              3  8   "   T F  qP                   x  K  	  R # 5ir   rE  r   s   & r!   r.   %cpp_dispatch_exprs.<locals>.<genexpr>  s     /hffhs   c                R   < V R 8g  ;'       g    SP                   P                  4       # )rn   )r  	is_out_fn)nr  s   &r!   rG  $cpp_dispatch_exprs.<locals>.<lambda>  s!    !u*::(8(8(::r#   r7   r   )r  r   r8   r   r:   rn  r   r   rR   r  __ne__)r  r  r  r  s   f$  r!   cpp_dispatch_exprsr    s    
 #15"A"K"K"MHE&(ABB/h//h// : 66
 ~~#fV]]E23Lr#   c          
     ,    V ^8  d   QhRRRRRRRRRR	/# )
r   r  r   rP   r   rQ   rS   rU   r   r   r   )r    s   "r!   r"   r"     sB     YJ YJYJ YJ0:YJGKYJYJr#   c                  VR JpV'       d    \        V 4      R(9  d   \        RV  R24      h\        V \        4      '       Ed   V P                  \
        P                  \
        P                  \
        P                  \
        P                  \
        P                  39   d%   V P                  P                  P                  4       # V P                  \
        P                  8X  d   V'       d   R# R# V P                  \
        P                  8X  d   V'       d   R# R# V P                  \
        P                  8X  d   R# V P                  \
        P                  8X  d   R# V P                  \
        P                   8X  d   V'       d   R# R# V P                  \
        P"                  8X  d   V'       d   R	# R
# V P                  \
        P$                  8X  d   R# V P                  \
        P                   8X  d   R# V P                  \
        P&                  8X  d   V'       d   R# R# V P                  \
        P(                  8X  d   R# EM\        V \*        4      '       d   \        V P,                  4      R8X  d   R# \        V P,                  4      R8X  d   R# \        V P,                  4      R8X  d   R# V'       g(   VR)9   d!   \/        V P,                  R R VR7      R,           # \/        V P,                  WVR7      # \        V \0        4      '       d   \        V P,                  4      R8X  d!   V P2                  e   RV P2                   R2# R# \        V P,                  4      R8X  d   R# \        V P,                  4      R8X  d   R# \        V P,                  4      R8X  d   R # \        V P,                  4      R!8X  d   R"# \        V P,                  4      R#8X  d   V'       d   R$# R # \        V P,                  4      R%8X  d   R&# \        RV  R'24      h)*Nztype 'z+' does not supported unpacking with defaultscalartypeWithDefault
scalartypedeviceWithDefaultrb  toInt64toSymInttoBoolWithDefaulttoBooltoDouble
stringViewlayoutWithDefaultra  memoryformatra   optionalTensorr,  	generatorz	Dimname[]toDimnameListOptionalrX   Optionalztensorlist_n<r  
tensorlistr%  list_of_optional_tensorsr&  dimnamelistr{   intlistr)  
doublelistr   
symintlistr#  
scalarlistz%' is not supported by PythonArgParser)ScalarType?r+  r/  Device?r.  Layout?rU   bool?)Nr[   r\   r]   )r   r   r8   r   r   r   ra   r2  r-  r#  r&  lowerr+  r/  r0  r{   r   rU   r)  r.  r1  r   rz   arg_parser_unpack_methodr   r   )r  rP   rQ   rS   has_default_inits   &&&$ r!   r  r    s    $4/CF 	+ 	 VA3&QRSS!X66MMMMNNMMNN
 
 66;;$$&&VVv(((.>*PLPVVv}}$*:&HHVVv)))VVvzz!VVv}}$!':6Y6VVv{{"*:&HHVVv||#VVvzz!VVv}}$*:&HHVVv***! + 
A|	$	$qvv;("#[K'[K'*!g 2
 '
 )tFKjX
 ,f  
Ax	 	 qvv;("010B]166(!,TT[I%-[I% [E![G#[H$#)<8y8[H$
s"GH
IIr#   c               (    V ^8  d   QhRRRRRRRR/# )	r   	arg_indexr{   r   rN   rS   rU   r   r   r   )r    s   "r!   r"   r"   p  s,      %26r#   c                   VP                   R Jp\        VP                  VP                  VP                   VR7      pV'       d   RVP                    2MRpRV RV  V R2p\	        VP
                  VV VR7      # )N)r  rP   rQ   rS   r'   r%   z_r.r&   r(   )r   r  r  r  )rQ   r  rO   rP   r   r   )r  r   rS   has_defaultunpack_methodrP   r  s   &&$    r!   arg_parser_output_exprr  p  s}     ..,K,
&&!))!..M (31>>"#GqG9A6D$VV	 r#   c               (    V ^8  d   QhRRRRRRRR/# )	r   r  r   r  r   rS   rU   r   z$dict[str, PythonArgParserOutputExpr]r   )r    s   "r!   r"   r"     s,      *7;)r#   c                   \        V P                  4       4       UUUu/ uF  w  r4\        W4VR 7      pVP                  VbK!  	  uppp# u upppi )rX   )	enumerater   r  r   )r  r  rS   ir   es   &&$   r!   arg_parser_output_exprsr    sL    
 blln--DA(f=A 	
	-  s   %Ar`  r  rb  r  ra  r  rc  r  rQ  c               (    V ^8  d   QhRRRRRRRR/# )	r   r  r   r  r   rS   rU   r   r
  r   )r    s   "r!   r"   r"     s2     y yy*y7;y yr#   c               
  aa \        WVR 7      p\        WVR 7      p. p/ o\        V4      pV P                  RR7       EF<  pVP                  pW7P                  ,          P
                  p	V'       d%   VR8X  d   VP                  RV	 R2.4       VSV&   KX  \        V\        4      '       d   \        VP                  4      ^8  dj   VP                  P                  4       '       dJ   VP                  RV	 R2.4       \        VP                  4       F  w  rRV
 R2SVP                  &   K  	  K  \        VP                  4      R	8X  d-   VP                  R
V RV	 R2RV RV RV R2.4       VSV&   EK7  V	SV&   EK?  	  V P                   '       d   RSR&   V P"                   Uu. uF  qwP                  NK  	  upoV'       Ed   VP                  P                  4       '       d   \%        VP                   R24      hV P"                   F  pVP                  \&        9  d&   \%        VP                   RVP                   R24      h\        VP                  4      \&        P)                  VP                  4      8w  g   Kv  \%        VP                   R\        VP                  4       RVP                   R24      h	  \*        ;QJ d#    V3R l\&         4       F  '       d   K   RM	  RM! V3R l\&         4       4      '       g   \%        VP                   RS 24      hVP-                  RVR,          P
                   RVR,          P
                   RVR,          P
                   RVR ,          P
                   R!VR",          P
                   R#24       R$SR$&   V'       Eg]   S'       EdT   RS9   Ed*   VP                  P                  4       '       g(   \%        VP                   R%V  R&V P                   24      h\*        ;QJ d    V3R' lR1 4       F  '       d   K   RM	  RM! V3R' lR1 4       4      '       g   \%        VP                   R(24      hVP-                  R)VR*,          P
                   R+VR,          P
                   R,VR,          P.                   R+VR,          P
                   R,VR,          P
                   R+VR,          P.                   R-24       R S9  d   \%        VP                   R.S R24      h\1        \2        ;QJ d    . V3R/ lV 4       F  NK  	  5VR07      # ! V3R/ lV 4       4      VR07      # u upi )2rX   T)r   r7   zauto self = ;zauto out = zout[r  z
Dimname[]?zauto __rv   z::std::optional<DimnameList> z = __z& ? ::std::make_optional(DimnameList(__z.value())) : ::std::nullopt;z : tensor options with output argz%: unrecognized tensor options field 'z' in python binding argumentsz: unrecognized type 'z' for tensor options field ''c              3  ,   <"   T F	  qS9   x  K  	  R # 5ir   r   r,   r   tensor_options_args_namess   & r!   r.   (dispatch_lambda_exprs.<locals>.<genexpr>  s     Q;Pa11;P   Fz": incomplete tensor options args: z0const auto options = TensorOptions()
    .dtype(r`  z)
    .device(rb  z)
    .layout(ra  z)
    .requires_grad(rQ  z)
    .pinned_memory(rc  z3);
torch::utils::maybe_initialize_device(options);
optionsz3: dtype in tensor_options_args without output arg, r_   c              3  ,   <"   T F	  qS9   x  K  	  R # 5ir   r   r   s   & r!   r.   r    s     T?S!55?Sr  z,: incomplete tensor options for output checkzcheck_out_type_matches(rn   r'   z,
                       z);
zE: expected "requires_grad" in tensor_options_args absent, but found [c              3  J   <"   T F  pSVP                   ,          x  K  	  R # 5ir   rE  )r,   r   lambda_args_exprss   & r!   r.   r    s     C{!%aff--{s    #)r  r  )ra  rb  )r  r  r  r   r   r  r   r8   ry   r3   r   r  r  r  r   rO   rR   r   r   TENSOR_OPTIONS_FIELDSre   r<   r   r  r
  r:   )r  r  rS   arg_parser_outputslambda_argsr  has_toptionsr   r   arg_parser_exprr  out_argr  r  s   &&$         @@r!   dispatch_lambda_exprsr    s    1vF&rV<KE(*%a(L \\d\3vv,VV499DFNLL"?"315
 '+d#q+,,AII"  ""LL!/!2!4
 (		2
481+!',,/ 3[L( LLdV3&7q93D6tfDjkojp  qM  N '+d# '6d#O 4T 
yyy$*&! 241G1G H1GA1G H|66!&&)IJKK''Avv22"vvhCAFF8Khi  166{377??"vvh3CK=@\]^]c]c\ddef  ( sQ;PQsssQ;PQQQ66(<=V<WX  	w',,- .)../ 0)../ 0&7<<= >&|499: ;
	
 (1)$ <55//66##%%"vvhQRTQUUVWYWcWcVde  3T?ST333T?STTT"vvhJK  LL*51667r:LW:U:Z:Z9[ \*73@@ADVW_D`DeDeCf g*8499:"=OPX=Y=f=f<g h ";;66(_`y_zz{|  'eC{Ce C{CC u !Is   $U )r   Nr   >   rU   voidr  r  void*
at::Scalar
at::Streamr  at::QSchemeat::ScalarTypeat::IntArrayRef::std::vector<at::Tensor>::std::tuple<double,int64_t>#::std::tuple<at::Tensor,at::Tensor>.::std::tuple<at::Tensor,at::Tensor,at::Tensor>2::std::tuple<at::Tensor,::std::vector<at::Tensor>>2::std::tuple<at::Tensor,at::Tensor,double,int64_t>6::std::tuple<at::Tensor,at::Tensor,at::Tensor,int64_t>9::std::tuple<at::Tensor,at::Tensor,at::Tensor,at::Tensor>=::std::tuple<at::Tensor,at::Tensor,double,at::Tensor,int64_t>A::std::tuple<at::Tensor,at::Tensor,at::Tensor,at::Tensor,int64_t>D::std::tuple<at::Tensor,at::Tensor,at::Tensor,at::Tensor,at::Tensor>O::std::tuple<at::Tensor,at::Tensor,at::Tensor,at::Tensor,at::Tensor,at::Tensor>r::std::tuple<at::Tensor,at::Tensor,at::Tensor,at::Tensor,c10::SymInt,c10::SymInt,at::Tensor,at::Tensor,at::Tensor>)=
__future__r   dataclassesr   typingr   torchgen.apir   torchgen.api.typesr   r   r	   torchgen.genr
   torchgen.modelr   r   r   r   r   r   r   r   r   r   collections.abcr   r   r@   rC   rN   ry   r   r   r   r   r   r   r
  r  r  rc   r8  r  r   rA  r~  rx   r  r  r   r  r  r  r  r  r  r  r  r  r  r   r#   r!   <module>r+     s   " !    G G *   2h?0 $      $Y) Y) Y)x $!C !C !CH $wQ wQ wQx $' ' 'f $   $'
 '
 '
^ $   $* * *0 $  2N N7=7$)=7:>=7@).<AF 	F
 F`.&EP$.4n
j)5^ ::U 04PYJNRYJ|9=&>B ]ii'W y>By yr#   