o h@sddlZddlmZddlmZddlmZmZmZddl Z ddl Z ddl Z ddl Z ddl Z ddlZ ddlZ ddlZ ddlmZedddZeddd ZGd d d e jjZdS) N)Sequence) nullcontext)AnyCallableOptional)torch_function_passthroughc Csttjftjtjjftjjtjjjftjtjjftjtjjftjtjjfg}itjjtjj tjj tjj tjj tjj tjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjj tjj tjj!tjj!tjj"tjj"tjj#tjj#tjj$tjj$tjj%tjj%tjj&tj'j(tjj)tj'j)i}|D]\}}|j*D]}|j+,|||j+,|<qqt-tjD]}|tjj*vrtjj+,||t.tj|<qtjj/j*D]}t.tj|dpt.t|}tjj/j+,|||<q|S)z} 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_attrr>h/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/_prims/context.pytorch_to_refs_mapsv                            r@cCsddtjjDS)zJ Set of all prim functions, e.g., torch._prims.add in all_prims() cSsh|] }tjj|qSr>)rr/r3r4).0r<r>r>r? Nszall_prims..)rr/r2r>r>r>r? all_primsIsrCc @sLeZdZdZdddefddZ  dd ed ed eed e e fd dZ dS) TorchRefsModeaB 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. FcGsdS)NFr>)_r>r>r?cszTorchRefsMode.cCs||_||_||_dS)N)strictshould_fallback_fnprims_mode_cls)selfrGrHrIr>r>r?__init__`s zTorchRefsMode.__init__r>N orig_functypesargskwargscCsl|duri}|tvs|tvr*|||i|WdS1s%wYt}||d}|durGt|tjjrGtj j |d}n-|durtt|tjj rtt |dd}|durh|j rht ||j dd}|durttj j |d}|dur||||||r||i|S|||i|WdS1swY|jrtdtj|||i|S)Ndefaultrzno _refs support for )rrCrIr@r4 isinstancer_ops OpOverload_decompdecomposition_tableOpOverloadPacketr6_dirrHrG RuntimeError overrides resolve_name)rJrLrMrNrOmappingfuncrPr>r>r?__torch_function__js6       z TorchRefsMode.__torch_function__)r>N) __name__ __module__ __qualname____doc__rrKrrrrdictr]r>r>r>r?rDQs" rD) functoolscollections.abcr contextlibrtypingrrrr torch._decomp torch._prims torch._refstorch._refs.nntorch._refs.nn.functionaltorch._refs.specialtorch.overridestorch._prims_commonr lru_cacher@rCrYTorchFunctionModerDr>r>r>r?s$    6