+ jhd^RIHt^RIt^RIHt^RIHtHtHt^RI H t ]'d ^RI H t H t ^RIt^RIt^RIt^RIt^RIt^RIt^RIt^RIt^RIHt] !R4t]!R 4t]P4R R l4t]P4R R l4t!RR]P:P<4tR#)) annotationsN) nullcontext)Any TYPE_CHECKINGTypeVar) ParamSpec)CallableSequence)torch_function_passthrough_P_RcV^8dQhRR/#)returnzdict[Any, Any])formats"U/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_prims/context.py __annotate__rs3 3 >3 c , \\P3\P\PP3\PP\PPP3\P\PP3\P \PP 3\P \PP 3.p/\PP\PPb\PP\PPb\PP\PPb\PP\PPb\PP \PP"b\PP$\PP&b\PP(\PP*b\PP,\PP.b\PP0\PP2b\PP4\PP6b\PP8\PP:b\PP<\PP<b\PP>\PP>b\PP@\PP@b\PPB\PPBb\PPD\PPDb\PPF\PPFb\PPH\PPH\PPJ\PPJ\PPL\PNPP\PPR\PNPR/CpVFNwr#VPTF9pVPVPYV4WPVPYV4&K; KP \[\P4FhpV\PPT9gK$\PPVPYV4V\]\PV4&Kj \PP^PTFkp\]\PVR4;'g\]\V4p\PP^PVPYV4W&Km V#)zq Mapping of torch API functions to torch._refs functions. E.g. torch_to_refs_map()[torch.add] == torch._refs.add N)0torch_refsnn functionalspecialfftlinalgTensor __invert__ bitwise_not__xor__ bitwise_xor__and__ bitwise_and__or__ bitwise_or__eq__eq__rsub__rsub __rtruediv__rtruediv __floordiv__ floor_divide __rfloordiv__ rfloordiv__pow__pow__rpow__rpow new_emptynew_full new_zerosnew_onesfill_zero_to sum_to_sizecopy__primscopy_toresize__all____dict__getdirgetattr _conversions)modulesr mod_torchmod_refss tensor_attrs rtorch_to_refs_maprMs   5;;>>"   ekknn778  ++, EKKOO$ u{{))* G !8!8 ekk55  ekk55  U[[33   U[[^^   u{{//   !!5;;#7#7  !!5;;#;#;  ""EKK$9$9  ekkoo  u{{//   5 5  u{{33   5 5  u{{33  EKK--!"  EKK--#$     %++"9"9 ELL00 U\\00-A0 ' !!A+3+<+<+@+@+CA  $$Q' (" '    ## #*/++*>*>*B*B1*EAgellA& ' [[ % % - -ellAt4IIq8I 11::>>qA. HrcV^8dQhRR/#)rrzset[Any]r)rs"rrrSsHH8Hrc\PPUu0uF,p\PPP V4kK. up#uupi)zB Set of all prim functions, e.g., torch._prims.add in all_prims() )rr>rArBrC)rKs r all_primsrPRs= 38,,2F2F G2FQELL ! ! % %a (2F GG Gs2AcB]tRt^ZtRtRR]3RRlltR RRlltR tR#) TorchRefsModea Switches the interpretation of torch.* functions and Tensor methods to use PrimTorch refs in torch._refs. (Direct calls to _refs are unaffected.) >>> # xdoctest: +SKIP >>> with TorchRefsMode(): ... torch.add(x, y) # calls torch._refs.add(x, y) By default, this context manager will fall back on the torch.* if the ref does not exist; set strict=True to error if this occurs. If the ref exists we still would like to fall back on the torch.* sometimes, this behavior can be customized by passing a function to should_fallback_fn. Fc R#)Fr)_s*rTorchRefsMode.lsUrc(V^8dQhRRRRRRRR/#) rstrictboolshould_fallback_fnzCallable[..., bool]prims_mode_clstyperNoner)rs"rrTorchRefsMode.__annotate__is2---0- -  -rc *WnW nW0nR#)N)rXrZr[)selfrXrZr[s&&&&r__init__TorchRefsMode.__init__is  "4,rNc ,V^8dQhRRRRRRRRR R /#) r orig_funczCallable[_P, _R]typeszSequence[type]argsz Sequence[Any]kwargszdict[str, Any] | Nonerrr)rs"rrr^ss<.*.*#.*.* .* & .* .*rc Vf/pV\9gV\49d,VP4;_uu_4V!V/VBuuRRR4#\4pVP VR4pVfV\ V\ PP4'd,\ PPP VR4pMVf\ V\ PP4'do\VRR4pVf0VP'd\WP^,R4pVe+\ PPP VR4pVe?VPWWcV4'd V!V/VB#V;_uu_4V!V/VBuuRRR4#VP'd,\!R\ P"P%V4 24hV!V/VB# +'giEL;i +'giLi;i)Ndefaultzno _refs support for )r rPr[rMrC isinstancer_ops OpOverload_decompdecomposition_tableOpOverloadPacketrE_dirrZrX RuntimeError overrides resolve_name)r`rdrerfrgmappingfuncris&&&&& r__torch_function__ TorchRefsMode.__torch_function__ss >F 2 2i9;6N$$&& $1&1'&$%{{9d+ >>!)^^A->E"}}88<rs""..'2  :t_ T] 3 3 l HHG*EOO55G*r