+
    &j3                       a  R t0 t ^ RIt^ RIHt ]P                  P                  t^ RIt^ RIt^ RIHt ^ RI	H
t
 ^ RIHt ^ RIHt / t] ^ k ]! 4       t] ^k ]
! R4      t]! R4      tR	 R
 ltR tRR R llt]! ]P.                  P0                  4      RR R ll4       t]! ]P.                  P4                  4      RR R ll4       tR# )    N)Tensor)Callable)TypeVar)	ParamSpec)Number_T_Pc                    V ^8  d   QhRR/# )   returnN )formats   "q/Users/jameslopez/projects/CWCArchive/cwc-podcast/.venv/lib/python3.14/site-packages/torch/jit/_decompositions.py__annotate__r      s     
L 
L4 
L    c                 R   \         P                  p\         P                  ! V 4      pVP                  P	                  4        F9  pVP
                  V8X  g   K  \        R VP                   RV P                   24      h	  VP                  V8X  d   \        RV P                   24      hR# )zNo signature on param z for function z"No return annotation for function N)	inspect_empty	signature
parametersvalues
annotationAssertionErrornamereturn_annotation)finspect_emptysigparams   &   r   (check_decomposition_has_type_annotationsr       s    NNM


A
C&&(}, (N166(K  ) -A!&&JKK .r   c                 *   V P                   pVP                   p\        V4      \        V4      8w  d   R # \        VP                  4       VP                  4       4       F  w  rE\        P
                  pR FP  pVR8X  d*   VP                  R8X  d   \        P                  ! R^R7       \        WG4      \        WW4      8w  g   KN    R # 	  VP                  pVP                  p	W8w  g   K  W8w  g   K  W8w  g   K   R # 	  V P                  VP                  8H  # )Fr   selfz4PyTorch uses 'input' instead of 'self' on public api)
stacklevel)r   r   )r   lenzipr   r   r   r   warningswarngetattrdefaultr   )
decomposition_sigtorch_op_sigdecomp_params	op_paramsdecomp_paramop_paramr   fielddecomp_default
op_defaults
   &&        r   signatures_matchr3   $   s    %00M''I
=S^+"%m&:&:&<i>N>N>P"Q
  +E<#4#4#>JWX |+wx/GG , &--%%
 *z/J++ #R. ..,2P2PPPr   c          
      $   V ^8  d   QhR\         P                  P                  R\        \        \         P
                  P                  3,          R,          R\        \        \        \        3,          .\        \        \        3,          3,          /# )r   aten_opregistryNr   )
torch_ops
OpOverloaddictstrjitScriptFunctionr   r	   r   )r   s   "r   r   r   E   sh     # #ZZ""#3		0001D8# xB (2r6"223#r   c                    a a R  V V3R llpV# )c                t    V ^8  d   QhR\         \        \        3,          R\         \        \        3,          /# )r   r   r   )r   r	   r   )r   s   "r   r   ,register_decomposition.<locals>.__annotate__I   s,      8BF#3 R8H r   c                   < Sf   \         o\        S\        P                  P                  4      '       g   \        R\        S4       24      hV P                  \        9   d   \        RV P                   24      h\        P                  V P                  4       \        P                  P                  V 4      p\        P                  P                  VP                  4       \        ^4       FU  p\        P                  P!                  VP                  4       \        P                  P#                  VP                  4       KW  	  VS\%        SP&                  4      &   V # )Nz'Expected aten_op to be OpOverload, got zDuplicated function name )decomposition_table
isinstancer7   r8   r9   r   type__name__function_name_setaddr<   script_C_jit_pass_inlinegraphrange_jit_pass_peephole_jit_pass_constant_propagationr;   _schema)r   scripted_func_r5   r6   s   &  r   decomposition_decorator7register_decomposition.<locals>.decomposition_decoratorI   s    *H'5::#8#899 9$w-I 
 ::** #<QZZL!IJJajj)		((+!!-"5"56qAHH''(;(;<HH33M4G4GH  *7W__%&r   r   )r5   r6   rR   s   ff r   register_decompositionrT   E   s     2 #"r   c          
          V ^8  d   QhR\         R\        \        ,          R,          R\        R,          R\        R\         /# )r   inputdimN
correctionkeepdimr   )r   listintr   bool)r   s   "r   r   r   i   sG     ! !!	cT	! ! 	!
 !r   c                 X   Vf   . pTp\        V\        \        34      '       d"   \        V4      ^ 8X  d   V P	                  4       pM$^pV F  pWPP
                  V,          ,          pK  	  \        P                  WR4      pW,
          pWw,          p\        P                  WV4      p	Vf   \        V^,
          4      p
M\\        V\        4      '       d   \        WR,
          4      p
M4\        V\        4      '       d   \        V4      V,
          p
M\        R4      hV	\        ^ V
4      ,          # )NTzcorrection must be int or float)rC   tuplerZ   r$   numelshapeatenmeansumfloatr[   RuntimeErrormax)rV   rW   rX   rY   dim_inrb   subsqrc   denoms   &&&&       r   var_decompositionrl   h   s     {#t}%%#c(a-KKMEU##A  99U&D
,C	B
((2G
$Ca!ej#&&!.)E
E**!Hz)E@AA Qr   c                <    V ^8  d   QhR\         R\        R\         /# )r   rV   unbiasedr   )r   r\   )r   s   "r   r   r      s&     G Gv G G Gr   c                 8    \        Y'       d   ^R7      # ^ R7      # )   )rX   )rl   )rV   rn   s   &&r   varrq      s    UXFF1FFr   c                    V ^8  d   Qh/ ^ \         9   d/   \        \        \        P                  P
                  3,          ;R&   ^\         9   d   \        \        ,          ;R&   # )r   rB   rF   )__conditional_annotations__r:   r;   r7   r<   r=   set)r   s   "r   r   r      sB     > =T#uyy7778 =  $ #3s8 # r   )N)NNF)T)rs   r7   r   opsra   r   r&   collections.abcr   typingr   typing_extensionsr   torch.typesr   rB   rt   rF   r   r	   r    r3   rT   rq   rX   rl   r)   r   )rs   s   @r   <module>rz      s      yy~~   $  '  <>  =!e  #T]t_
LQB#F ++,! -!H (()G *Gr   