Home
last modified time | relevance | path

Searched refs:get_default_nowrap_functions (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch_function.py8 from torch.overrides import _get_overloaded_args, get_default_nowrap_functions
51 for fn in get_default_nowrap_functions()
H A Dtorch.py385 @register(torch.overrides.get_default_nowrap_functions.__wrapped__)
394 tx, torch.overrides.get_default_nowrap_functions()
/aosp_15_r20/external/pytorch/torch/masked/maskedtensor/
H A Dcore.py9 from torch.overrides import get_default_nowrap_functions
297 if func in get_default_nowrap_functions():
/aosp_15_r20/external/pytorch/torch/
H A D_tensor.py22 get_default_nowrap_functions,
1513 if func in get_default_nowrap_functions():
H A Doverrides.py377 def get_default_nowrap_functions() -> Set[Callable]: function