IЦigSSKrSSKJr SSKJrJrJrJrJr SSK r SSK r SSK r SSK r SSK r SSKr SSKr SSKr SSKJr \R&"S5S5r\R&"S5S5r"SS\ R,R.5rg) N) nullcontext)AnyCallableDictOptionalSequence)torch_function_passthroughc ( [[R4[R[RR4[RR[RRR4[R[RR4[R [RR 4[R [RR 4/n0[RR[RR_[RR[RR_[RR[RR_[RR[RR_[RR [RR"_[RR$[RR&_[RR([RR*_[RR,[RR._[RR0[RR2_[RR4[RR6_[RR8[RR:_[RR<[RR<_[RR>[RR>_[RR@[RR@_[RRB[RRB_[RRD[RRD_[RRF[RRF_[RRH[RRH[RRJ[RRJ[RRL[RNRP[RRR[RNRR0EnUHNup#URTH9nURVRYU5XRVRYU5'M; MP [[[R5HgnU[RRT;dM#[RRVRYU5U[][RU5'Mi [RR^RTHjn[][RUS5=(d [][U5n[RR^RVRYU5X'Ml U$)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 S/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/_prims/context.pytorch_to_refs_maprBs   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& ' [[ % % - -ellAt4Iq8I 11::>>qA. Hc[RRVs1sH,n[RRR U5iM. sn$s snf)zB Set of all prim functions, e.g., torch._prims.add in all_prims() )r r2r5r6r7)r?s rA all_primsrEHs= 38,,2F2F G2FQELL ! ! % %a (2F GG Gs3Ac X\rSrSrSrSS\4SjrS S\S\S \\ S \ \ 4S jjr S r g) TorchRefsModePa 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. Fcg)NF)_s rATorchRefsMode.bserCc(XlX lX0lg)N)strictshould_fallback_fnprims_mode_cls)selfrOrPrQs rA__init__TorchRefsMode.__init___s  "4,rCN orig_functypesargskwargscnUc0nU[;dU[5;a"UR5 U"U0UD6sSSS5 $[5nUR US5nUcT[ U[ RR5(a+[ RRR US5nOfUcc[ U[ RR5(a:[USS5nUb*[ RRR US5nUb4URXXcU5(aU"U0UD6$U U"U0UD6sSSS5 $UR(a+[S[ R R#U535eU"U0UD6$!,(df  GNc=f!,(df  Nd=f)Ndefaultzno _refs support for )r rErQrBr7 isinstancer _ops OpOverload_decompdecomposition_tableOpOverloadPacketr9rPrO RuntimeError overrides resolve_name)rRrUrVrWrXmappingfuncrZs rA__torch_function__ TorchRefsMode.__torch_function__isr >F 2 2i9;6N$$& $1&1'&#%{{9d+ F& F#& F4)rQrPrO)rJN)__name__ __module__ __qualname____firstlineno____doc__rrSrrrrrrf__static_attributes__rJrCrArGrGPsW  +" -!!% )*)*)*sm )*  )*)*rCrG) functools contextlibrtypingrrrrrr torch._decomp torch._prims torch._refstorch._refs.nntorch._refs.nn.functionaltorch._refs.specialtorch.overridestorch._prims_commonr lru_cacherBrErbTorchFunctionModerGrJrCrAr{s"::  : T3 3 l THHB*EOO55B*rC