
    [ThTk                     N   S SK r S SKrS SKrS SKrS SKJrJr  S SKrS SKJr  \R                   " S S5      5       r
\
" 5       r      SS jrS\\\4   4S jr\ R                   S	 5       rS
 r " S S5      rSS jrSS jrSS jrS rS rS rSS.S jrSS.S jrSS.S jrg)    N)AnyOptional)infc                   b    \ rS rSr% Sr\\S'   Sr\\S'   Sr	\\S'   Sr
\\S	'   S
r\\   \S'   Srg
)__PrinterOptions      	precision  	threshold   	edgeitemsP   	linewidthNsci_mode )__name__
__module____qualname____firstlineno__r
   int__annotations__r   floatr   r   r   r   bool__static_attributes__r       I/var/www/auris/envauris/lib/python3.13/site-packages/torch/_tensor_str.pyr   r      s8    IsIuIsIs#Hhtn#r   r   c                    Ub  US:X  a-  S[         l        S[         l        S[         l        S[         l        OiUS:X  a-  S[         l        S[         l        S[         l        S[         l        O6US	:X  a0  S[         l        [
        [         l        S[         l        S[         l        U b  U [         l        Ub  U[         l        Ub  U[         l        Ub  U[         l        U[         l        g)
a  Set options for printing. Items shamelessly taken from NumPy

Args:
    precision: Number of digits of precision for floating point output
        (default = 4).
    threshold: Total number of array elements which trigger summarization
        rather than full `repr` (default = 1000).
    edgeitems: Number of array items in summary at beginning and end of
        each dimension (default = 3).
    linewidth: The number of characters per line for the purpose of
        inserting line breaks (default = 80). Thresholded matrices will
        ignore this parameter.
    profile: Sane defaults for pretty printing. Can override with any of
        the above options. (any one of `default`, `short`, `full`)
    sci_mode: Enable (True) or disable (False) scientific notation. If
        None (default) is specified, the value is defined by
        `torch._tensor_str._Formatter`. This value is automatically chosen
        by the framework.

Example::

    >>> # Limit the precision of elements
    >>> torch.set_printoptions(precision=2)
    >>> torch.tensor([1.12345])
    tensor([1.12])
    >>> # Limit the number of elements shown
    >>> torch.set_printoptions(threshold=5)
    >>> torch.arange(10)
    tensor([0, 1, 2, ..., 7, 8, 9])
    >>> # Restore defaults
    >>> torch.set_printoptions(profile='default')
    >>> torch.tensor([1.12345])
    tensor([1.1235])
    >>> torch.arange(10)
    tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

Ndefaultr	   r   r   r   short   full)
PRINT_OPTSr
   r   r   r   r   r   )r
   r   r   r   profiler   s         r   set_printoptionsr%      s    Z i#$J #'J #$J #%J #$J #'J #$J #%J #$J #&J #$J #%J (
(
(
(
"Jr   returnc                  6    [         R                  " [        5      $ )zqGets the current options for printing, as a dictionary that
can be passed as ``**kwargs`` to set_printoptions().
)dataclassesasdictr#   r   r   r   get_printoptionsr*   b   s     j))r   c               +   v   #    [        5       n[        S0 U D6   Sv   [        S0 UD6  g! [        S0 UD6  f = f7f)zuContext manager that temporarily changes the print options.  Accepted
arguments are same as :func:`set_printoptions`.Nr   )r*   r%   )kwargs
old_kwargss     r   printoptionsr.   i   s9      "#Jv'&:&&:&s   9) 969c                    U R                   (       dI  U R                  (       aH  [        R                  R	                  U R
                  5      R                  (       d  [        R                  O[        R                  nU R                  US9$ )N)dtype)
is_mpsis_xputorchxpuget_device_propertiesdevicehas_fp64r   doubleto)tr0   s     r   tensor_totyper;   u   sW     HH!@!@!J!S!S 	
 \\ 
 44e4r   c                   &    \ rS rSrS rS rS rSrg)
_Formatter   c                 x   UR                   R                  U l        SU l        SU l        SU l        [        R                  " 5          UR                  S5      nS S S 5        U R                  (       d2  W H*  nU n[        U R
                  [        U5      5      U l        M,     GO_[        R                  " W[        R                  " U5      UR                  S5      -  5      nUR                  5       S:X  a  g UR                   [        R                  :X  a  UR!                  5       n[#        UR%                  5       5      n[#        UR'                  5       5      n[#        UR                  5       5      nU H&  nU[        R(                  " U5      :w  d  M  SU l          O   U R                  (       a  X-  S:  d  US:  aX  SU l        U HJ  nS[*        R,                   S	3R/                  U5      n[        U R
                  [        U5      5      U l        ML     OU H.  nUS
 n[        U R
                  [        U5      S-   5      U l        M0     OX-  S:  d  US:  d  US:  aX  SU l        U HJ  nS[*        R,                   S	3R/                  U5      n[        U R
                  [        U5      5      U l        ML     OPU HJ  nS[*        R,                   S3R/                  U5      n[        U R
                  [        U5      5      U l        ML     [*        R                  b  [*        R                  U l        g g ! , (       d  f       GN= f)NTF   r   g     @@g    חA{:.e}.0fg-C6?f})r0   is_floating_pointfloating_dtypeint_moder   	max_widthr3   no_gradreshapemaxlenmasked_selectisfinitenenumelfloat8_e8m0fnur   r;   absminceilr#   r
   format)	selftensortensor_viewvalue	value_strnonzero_finite_valsnonzero_finite_absnonzero_finite_minnonzero_finite_maxs	            r   __init___Formatter.__init__   s   $ll<<]]_ ..,K  ""$$g	!$T^^S^!D %
 #("5"5U^^K8;>>!;LL# #((*a/||u333 ':&?&?&A# "//B/F/F/H!I!./A/E/E/G!H!./A/E/E/G!H,EJJu--$)DM -
 }} ';fD)E1$(DM!4&*:+?+?*@$D$K$KE$R	),T^^S^)L "5 "5',Sk	),T^^S^a=O)P "5 ';fD)E1)F2$(DM!4&*:+?+?*@$D$K$KE$R	),T^^S^)L "5 "5&*:+?+?*@$D$K$KE$R	),T^^S^)L "5 *&//DM +I _s   L**
L9c                     U R                   $ N)rI   )rW   s    r   width_Formatter.width   s    ~~r   c                    U R                   (       a  U R                  (       a1  SU R                   S[        R                   S3R                  U5      nOxU R                  (       a@  US n[        R                  " U5      (       d   [        R                  " U5      (       d  US-  nO'S[        R                   S3R                  U5      nOU nU R                  [        U5      -
  S-  U-   $ )Nz{:.rC   rD   rB   rE    )rG   r   rI   r#   r
   rV   rH   mathisinfisnanrM   )rW   rZ   rets      r   rV   _Formatter.format   s    }}DNN+1Z-A-A,B#FMMeTs

5))TZZ->->3JCZ112#6==eDGCS)S0366r   )rG   rH   rI   r   N)r   r   r   r   r`   rd   rV   r   r   r   r   r=   r=      s    K0Z7r   r=   c                     Ub[  [        U R                  U5      n[        U R                  U5      S-   R                  5       nUS   S:X  d	  US   S:X  a  X4-   $ US-   U-   $ UR	                  U R                  5       5      $ Njr   +-)_scalar_strrealimaglstriprV   item)rW   
formatter1
formatter2real_strimag_strs        r   rs   rs      s|    tyy*5		:6<DDFA;#!!3&&c>H,,  --r   c           	         UR                  5       S-   nUb  XTR                  5       S-   -  n[        S[        [        R                  " [
        R                  U-
  U-  5      5      5      nX44S jnU(       a  [
        R                  (       d  S/nOU(       a  U R                  S5      S[
        R                  -  :  au  U S [
        R                   R                  5        V	s/ s H
  o" U	5      PM     sn	S/-   U [
        R                  * S  R                  5        V	s/ s H
  o" U	5      PM     sn	-   nO%U R                  5        V	s/ s H
  o" U	5      PM     nn	[        S[        U5      U5       V
s/ s H	  oXU-    PM     nn
U Vs/ s H  nSR                  U5      PM     nnSS	S
US-   -  -   R                  U5      -   S-   $ s  sn	f s  sn	f s  sn	f s  sn
f s  snf )Nr!   r@   c                     Ube  UR                  U R                  5      nUR                  U R                  5      S-   R                  5       nUS   S:X  d	  US   S:X  a  X4-   $ US-   U-   $ UR                  U 5      $ ro   )rV   rt   ru   rv   )valrx   ry   rz   r{   s        r   _val_formatter#_vector_str.<locals>._val_formatter   s~    !!((2H"))#((3c9AACH{c!Xa[C%7**#~00$$S))r   ...r   z ..., [,
rh   ])rd   rL   r   ri   floorr#   r   r   sizetolistrangerM   join)rW   indent	summarizerx   ry   element_lengthelements_per_liner   datar~   i
data_lineslineliness                 r   _vector_strr      s   %%'!+N**,q00	3tzz://&8^LMN (2 
* --w	tyy|a**>*>&>>,01G:3G3G,H,O,O,QR,QS^C ,QRh.2J4H4H3H3J.K.R.R.TU.Ts~c".TUV 	 04{{}=}s#}= 27q#d)EV1W1WAQ&&'1W   *44TYYt_E4*sfqj1177>>DD SU > 5s   GGG.GGc                    U R                  5       nUS:X  a  [        XU5      $ US:X  a  [        XX#U5      $ U(       a  U R                  S5      S[        R
                  -  :  a  [        S[        R
                  5       Vs/ s H  n[        X   US-   X#U5      PM     snS/-   [        [        U 5      [        R
                  -
  [        U 5      5       Vs/ s H  n[        X   US-   X#U5      PM     sn-   nO<[        SU R                  S5      5       Vs/ s H  n[        X   US-   X#U5      PM     nnSSUS-
  -  -   SUS-   -  -   R                  U5      nSU-   S	-   $ s  snf s  snf s  snf )
Nr   r@   r!   r   ,
rh   r   r   )
dimrs   r   r   r#   r   r   _tensor_str_with_formatterrM   r   )	rW   r   r   rx   ry   r   r   slices
tensor_strs	            r   r   r     s   
((*C
ax4Z88
ax4
KKTYYq\A
(<(<$<< q*"6"67	 8A +GVaZ
 8	 g s4y:+?+??TK	 LA +GVaZ
 L	 	( 1diil+	
 , '!YJ ,	 	 
 a((3&1*+==CCFKJc!!/
s   ?EE"E'c                    U R                  5       S:X  a  gU R                  5       (       a  U R                  S 5      n U R                  5       [        R                  :  nU R                  5       (       a  U R                  5       n U R                  5       (       a  U R                  5       n U R                  [        R                  [        R                  [        R                  [        R                  4;   a  U R                  5       n U R                  R                   (       a  U R#                  5       n [%        U(       a  ['        U R(                  5      OU R(                  5      n[%        U(       a  ['        U R*                  5      OU R*                  5      n[-        XX#U5      $ [%        U(       a  ['        U 5      OU 5      n[-        XX%5      $ )Nr   [])rQ   	has_namesrenamer#   r   _is_zerotensorcloneis_negresolve_negr0   r3   float8_e5m2float8_e5m2fnuzfloat8_e4m3fnfloat8_e4m3fnuzhalf
is_complexresolve_conjr=   get_summarized_datart   ru   r   )rW   r   r   real_formatterimag_formatter	formatters         r   _tensor_strr   @  sS   zz|q~~ {{4 

z333Izz| {{}}! zz	  yy{zz  "#.7		*TYY
 $.7		*TYY
 *)^
 	
 I2484P	)$	MMr   c                 n   U /n[        U 5      U R                  S5      -
  S-   nU Hk  n[        U5      nU(       d  XW-   S-   [        R                  :  a"  UR	                  SSU-  -   U-   5        X'-   nSnMP  UR	                  SU-   5        XWS-   -  nMm     UR	                  S5        S	R                  U5      $ )
Nr   r@   r!   r   rh   Fr   ) )rM   rfindr#   r   appendr   )r   suffixesr   force_newlinetensor_strslast_line_lensuffix
suffix_lens           r   _add_suffixesr   o  s    ,K
Oj&6&6t&<<q@M[
M6:Z=Q=QQusV|3f<="/M!Mtf}-!^+M  s77;r   c                    U R                  5       nUS:X  a  U $ US:X  ac  U R                  S5      S[        R                  -  :  a;  [        R
                  " U S [        R                   U [        R                  * S  45      $ U $ [        R                  (       d#  U R                  S/U R                  5       -  5      $ U R                  S5      S[        R                  -  :  a  [        S[        R                  5       Vs/ s H  o U   PM	     nn[        [        U 5      [        R                  -
  [        U 5      5       Vs/ s H  o U   PM	     nn[        R                  " X4-    Vs/ s H  n[        U5      PM     sn5      $ [        R                  " U  Vs/ s H  n[        U5      PM     sn5      $ s  snf s  snf s  snf s  snf )Nr   r@   r!   )r   r   r#   r   r3   cat	new_emptyr   rM   stackr   )rW   r   r   startendxs         r   r   r     sj   
((*C
ax
ax99Q<!j222299,
,,-tZ5I5I4I4K/LM  K~~qcDHHJ.//	1J000	0"':+?+?"@A"@Qa"@A %c$i*2F2F&FD	 RS R1Aw RS{{U[J[q/2[JKK{{DADq/2DABB	 BSJAs   <F/=F4#F9F>tensor_contentsc          	        ^!^" [         R                  R                  R                  U 5      (       a	  [	        XS9$ [        U 5      [         R                  L =(       d%    [        U 5      [         R                  R                  L nU R                  (       a  SnO"U(       a  SnO[        U 5      R                   S3n[        U5      m!/ nUS LnU(       a  Un[         R                  R                  R                  U 5      u  pxUR                  R
                  [         R                  R!                  5       :w  dj  UR                  R
                  S:X  a6  [         R"                  R%                  5       UR                  R&                  :w  d  UR                  R
                  S:X  a*  UR)                  S[+        UR                  5      -   S-   5        UR                  R
                  S	;   a  UR-                  S
5      n[         R.                  " 5       [         R0                  :X  a  [         R2                  O[         R4                  n	UR6                  [         R.                  " 5       U	[         R8                  [         R:                  4;   n
UR<                  (       Ga  UR)                  S[+        [?        UR@                  5      5      -   5        SSK!J"n  URF                  =(       d    [I        X{5      nU(       d+  UR)                  S[+        URK                  5       5      -   5        U
(       d'  UR)                  S[+        UR6                  5      -   5        U(       Gd  SnURM                  5       RO                  5       nU(       a  SnO[Q        UT![        U5      -   5      nU(       d  URS                  5       S:X  a$  US[+        [?        UR@                  5      5      -   -  nSnURU                  5       RO                  5       nU(       a  SnO[Q        UT![        U5      -   5      nU(       d  URS                  5       S:X  a$  US[+        [?        UR@                  5      5      -   -  nUU-   S-   ST!-  -   U-   U-   S-   nG	OURV                  [         RX                  [         RZ                  [         R\                  [         R^                  1;   Ga  SSK!J"n  UR)                  S[+        [?        UR@                  5      5      -   5        URF                  =(       d    [I        X{5      nU(       d+  UR)                  S[+        URK                  5       5      -   5        U
(       d'  UR)                  S[+        UR6                  5      -   5        U(       Gd  [         RX                  [         R                  R`                  [         R                  Rb                  4[         RZ                  [         R                  Rd                  [         R                  Rf                  4[         R\                  [         R                  R`                  [         R                  Rb                  4[         R^                  [         R                  Rd                  [         R                  Rf                  40URV                     u  nnURV                  [         RX                  [         R\                  1;   a  Su  nnOSu  nnSUS S  S3nU" U5      RO                  5       nU(       a  SnO[Q        UT![        U5      -   5      nURS                  5       S:X  d  U(       a$  US[+        [?        UR@                  5      5      -   -  nUS S  S3nU" U5      RO                  5       nU(       a  SnO[Q        UT![        U5      -   5      nURS                  5       S:X  d  U(       a$  US[+        [?        UR@                  5      5      -   -  nSnURi                  5       RO                  5       nU(       a  SnO[Q        UT![        U5      -   5      nURS                  5       S:X  d  U(       a$  US[+        [?        UR@                  5      5      -   -  nUU-   S-   ST!-  -   U-   U-   S-   ST!-  -   U-   U-   S-   nGOURj                  (       Ga.  UR)                  S[+        [?        UR@                  5      5      -   5        U
(       d'  UR)                  S[+        UR6                  5      -   5        UR)                  S[+        URm                  5       5      -   5        URm                  5       [         Rn                  :X  d"  URm                  5       [         Rp                  :X  aW  UR)                  S[+        URs                  5       5      -   5        UR)                  S[+        URu                  5       5      -   5        OURm                  5       [         Rv                  :X  dD  URm                  5       [         Rx                  :X  d"  URm                  5       [         Rz                  :X  a  UR)                  S[+        UR}                  5       5      -   5        UR)                  S[+        UR                  5       5      -   5        UR)                  S[+        UR                  5       5      -   5        U(       d  [Q        UR                  5       T!5      nGOUR                  (       a`  U(       dW  S  m"S!R                  U!U"4S" j[         R                  R                  R                  R                  US5       5       5      nS#U S$3nGOf[         R                  " U5      (       a#  S%n[        [         R                  " U5      5      nGO(SSK!J"n  URF                  (       d  [I        X{5      (       a  UR)                  S[+        [?        UR@                  5      5      -   5        UR6                  [         R.                  " 5       :w  a'  UR)                  S[+        UR6                  5      -   5        U(       d  SnGO|URS                  5       S:X  a  UR<                  (       d  UR                  5       S&:w  a0  UR)                  S[+        [?        UR@                  5      5      -   5        UR6                  [         R.                  " 5       :w  a'  UR)                  S[+        UR6                  5      -   5        U(       d  S'nO[        R                  (       d0  UR)                  S[+        [?        UR@                  5      5      -   5        U
(       d'  UR)                  S[+        UR6                  5      -   5        U(       dE  URV                  [         R                  :w  a  [Q        UR                  5       T!5      nO[Q        UT!5      nURV                  [         R                  :w  a'  UR)                  S([+        URV                  5      -   5        S n U R                  nUcA  Wb>  [        U5      R                  nUS*:X  a#  UR                  5       R                  S+S&5      S,   nUb  UR)                  S-U S.35        O"U R                  (       a  UR)                  S/5        UR                  5       (       a  UR)                  S0UR                   35        Ub  UR)                  S1U 35        [        UW-   UT!UR<                  S29n [I        U[         R                  R                  5      (       a  U(       d  S3U  S3n U $ ! [         a    S)n GN)f = f)4Nr   znested_tensor(ztensor((cudampszdevice='')xlalazyipumtiacpuzsize=r   )
FakeTensorznnz=zdtype=zindices=tensor(r   z, size=zvalues=tensor(z),
rh   r   )rowcolumn)r   r   cr   z_indices=tensor(zquantization_scheme=zscale=zzero_point=zaxis=c                 P    SR                  S U R                  S5       5       5      $ )Nr   c              3   ,   #    U  H
  nS U 3v   M     g7f)z  Nr   ).0r   s     r   	<genexpr>4_str_intern.<locals>.indented_str.<locals>.<genexpr>J  s      G2dVs   )r   split)sr   s     r   indented_str!_str_intern.<locals>.indented_strI  s    yy G GGGr   r   c              3   N   >#    U  H  nT" [        U5      TS -   5      v   M     g7f)r@   N)str)r   r:   r   r   s     r   r   _str_intern.<locals>.<genexpr>L  s*      ;A SVVaZ00;s   "%z[
z
]z_to_functional_tensor(r@   r   zlayout=InvalidCppFunctionz::rA   z	grad_fn=<>zrequires_grad=Trueznames=ztangent=)r   z
Parameter()Wr3   _C
_functorchis_functorch_wrapped_tensor_functorch_wrapper_str_interntypeTensornn	Parameter	is_nestedr   rM   autograd
forward_adunpack_dualr6   _get_default_devicer   current_deviceindexr   r   r9   get_default_dtyper8   cdoublecfloatr0   int64r   	is_sparsetupleshapetorch._subclasses.fake_tensorr   is_meta
isinstance_nnz_indicesdetachr   rQ   _valueslayout
sparse_csr
sparse_csc
sparse_bsr
sparse_bsccrow_indicescol_indicesccol_indicesrow_indicesvaluesis_quantizedqschemeper_tensor_affineper_tensor_symmetricq_scaleq_zero_pointper_channel_affineper_channel_symmetric per_channel_affine_float_qparamsq_per_channel_scalesq_per_channel_zero_pointsq_per_channel_axis
dequantizer   opsatenunbindr   _is_functional_tensorrepr_from_functional_tensorr   r#   r   stridedto_densegrad_fnRuntimeErrornamersplitrequires_gradr   namesr   )#inpr   is_plain_tensorprefixr   custom_contents_providedr   rW   tangent_default_complex_dtypehas_default_dtyper   r   indices_prefixindicesindices_strvalues_prefixr  
values_strcompressed_indices_methodplain_indices_methodcdimnamepdimnamecompressed_indices_prefixcompressed_indicescompressed_indices_strplain_indices_prefixplain_indicesplain_indices_strstrsgrad_fn_namer  string_reprr   r   s#                                    @@r   _str_internr=    sb   xx66s;;,SRR3i5<</R49@R@R3RO
}}!	I&&'q)[FH.d:$

 NN--99#>MD 	EHH88::KK&

))+t{{/@/@@KK%
S%55;<
 {{99wwu~ 002ellB  

!

	'  ~~~#eDJJ&7"889<,,>*T">OOFS%556 OOHs4::67'.Nmmo,,.G#)'6C<O3OP'--/Q.y3uW]]/C+DDD,M\\^**,F"
(#m:L1LM
&,,.A-i#eFLL.A*BBB
 ,  	 
    
	 
 	=#eDJJ&7"889,,>*T">OOFS%556 OOHs4::67'  5<<#<#<ell>V>V"W  5<<#<#<ell>V>V"W  5<<#<#<ell>V>V"W  5<<#<#<ell>V>V"W	?
 kk?;%'; {{u//1A1ABB%4"(%4"(*+HRaL>9I(J%!:4!@!G!G!I).&)4&5N1O(O*& "'')Q.'&)c,2237 + & '/rl^3C#D 06==?M$)!$/!6C0D,E#E%! ""$)W!YU=;N;N5O1P%PP!,M[[]))+F"
(#m:L1LM
||~"gi#eFLL.A*BBB
)() , '	'
 $$  ,    	 
  
			#eDJJ&7"889 OOHs4::67.T\\^1DDELLNe555||~!;!;;OOHs4<<>'::;OOMC0A0A0C,DDELLNe666||~!<!<<||~!G!GGOOHs4+D+D+F'GGHOOMC0N0N0P,QQROOGc$*A*A*C&DDE'$T__%6?J	'H :: ..224; D tfCJ		$	$T	*	*)%77=>
 	=<<:d77OOGc%

*;&<<=zzU44663tzz? :; ,"
zz|q 88:?OOGc%

2C.D$DE ::!8!8!::OOHs4::$>?/!%J!++OOGc%

2C.D$DE(OOHs4::$>?/{{emm3%0&%I
%0v%>
{{emm#	C$445 L!++  3G}--=("<<>00q9"=L)L>34			,-~~&-.(7),-nn	K $**++O";-q1I  ! !	!s   {, ,{<;{<c                J   [         R                  R                  R                  U 5      nUS:w  d   e[         R                  R                  R	                  U 5      (       a  [         R
                  " U 5        [         R                  R                  R                  U 5      n[        U5      n[        R                  " US5      n[         R                  R                  R                  U 5      (       a=  [         R                  R                  R                  U 5      nUS:w  d   eSU SU SU S3$ [         R                  R                  R                  U 5      (       a	  SU SU S3$ [         R                  R                  R	                  U 5      (       a	  SU S	U S
3$ [        S5      e)NrA   z    zBatchedTensor(lvl=z, bdim=z	, value=
z
)zGradTrackingTensor(lvl=zFunctionalTensor(lvl=z
, value=\
r   z8We don't know how to print this, please file us an issue)r3   r   r   maybe_get_levelis_functionaltensor_syncget_unwrappedr  textwrapr   is_batchedtensormaybe_get_bdimis_gradtrackingtensor
ValueError)rX   r   levelrZ   
value_reprindented_value_reprbdims          r   r   r     sV   HH//7EB;;xx..v66 	FHH--f5EeJ"//*g>xx++F33xx""11&9rzz wtfJ?R>SSVW	
 xx0088(z:M9NcRRxx..v66&ug\*QGG
O
PPr   c                H   [         R                  " 5          [         R                  R                  R	                  5          [         R
                  R                  5       n[        XS9sS S S 5        sS S S 5        $ ! , (       d  f       O= fS S S 5        g ! , (       d  f       g = f)Nr   )r3   rJ   utils_python_dispatch_disable_current_modesr   _DisableFuncTorchr=  )rW   r   guards      r   _strrR    sU    	%++66MMO**,4A POOOs"   )B'A9&	B9
B	B
B!)NNNNNNrc   )
contextlibr(   ri   rC  typingr   r   r3   r   	dataclassr   r#   r%   dictr   r*   contextmanagerr.   r;   r=   rs   r   r   r   r   r   r=  r   rR  r   r   r   <module>rX     s            $ $ $ 

 F#R*$sCx. * ' '	]7 ]7@
.'EZ""J,N^  C* )- Qh >B Q6 #' Br   