Home
last modified time | relevance | path

Searched refs:_detect_infra_mode (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/_functorch/_aot_autograd/
Ddispatch_and_compile_graph.py19 from torch.utils._python_dispatch import _detect_infra_mode
138 proxy_mode = _detect_infra_mode(torch._C._TorchDispatchModeKey.PROXY)
Dtraced_function_transforms.py238 functional_tensor_mode = torch.utils._python_dispatch._detect_infra_mode(
684 functional_tensor_mode = torch.utils._python_dispatch._detect_infra_mode(
/external/pytorch/torch/_subclasses/
Dfunctional_tensor.py15 _detect_infra_mode,
244 functional_mode = _detect_infra_mode(torch._C._TorchDispatchModeKey.FUNCTIONAL)
281 if _detect_infra_mode(torch._C._TorchDispatchModeKey.FUNCTIONAL).export:
/external/pytorch/torch/utils/
D_python_dispatch.py134 def _detect_infra_mode(key): function