| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | opinfo_schema.py | 4 from torch._dispatch.python import enable_python_dispatcher
99 with enable_python_dispatcher():
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/ |
| H A D | shape_prop.py | 7 from torch._dispatch.python import enable_python_dispatcher
151 with self.fake_mode, enable_python_dispatcher():
|
| /aosp_15_r20/external/pytorch/torch/_dispatch/ |
| H A D | python.py | 17 enable_python_dispatcher = torch._C._EnablePythonDispatcher variable
174 with enable_python_dispatcher(), unittest.mock.patch(
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_efficient_conv_bn_eval.py | 101 from torch._dispatch.python import enable_python_dispatcher
136 with enable_python_dispatcher():
|
| H A D | test_torchinductor_opinfo.py | 14 from torch._dispatch.python import enable_python_dispatcher
685 with enable_python_dispatcher():
|
| H A D | test_torchinductor.py | 32 from torch._dispatch.python import enable_python_dispatcher
11418 with enable_python_dispatcher():
|
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | dispatch_and_compile_graph.py | 14 from torch._dispatch.python import enable_python_dispatcher
48 with enable_python_dispatcher(), FunctionalTensorMode(
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 14 from torch._dispatch.python import enable_python_dispatcher
569 ) as mode, enable_python_dispatcher():
700 ), enable_python_dispatcher():
978 ) as mode, enable_python_dispatcher():
997 ) as mode, enable_python_dispatcher():
1016 ) as mode, enable_python_dispatcher():
|
| H A D | test_prims.py | 341 with torch._dispatch.python.enable_python_dispatcher():
409 with torch._dispatch.python.enable_python_dispatcher():
|
| H A D | test_proxy_tensor.py | 165 from torch._dispatch.python import enable_python_dispatcher
166 with enable_python_dispatcher():
462 from torch._dispatch.python import enable_python_dispatcher
464 with enable_python_dispatcher():
488 from torch._dispatch.python import enable_python_dispatcher
490 with enable_python_dispatcher():
|
| H A D | test_functionalization.py | 9 enable_python_dispatcher,
1914 with enable_python_dispatcher():
2038 with enable_python_dispatcher():
|
| H A D | test_meta.py | 13 from torch._dispatch.python import enable_python_dispatcher
609 with enable_python_dispatcher():
1723 with enable_python_dispatcher():
|
| /aosp_15_r20/external/executorch/exir/verification/ |
| H A D | verifier.py | 22 from torch._dispatch.python import enable_python_dispatcher
168 with enable_python_dispatcher(), fake_mode:
|
| /aosp_15_r20/external/executorch/exir/capture/ |
| H A D | _capture.py | 28 from torch._dispatch.python import enable_python_dispatcher
303 with enable_python_dispatcher(), fake_tensor_mode:
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | aot_autograd.py | 16 from torch._dispatch.python import enable_python_dispatcher
579 enable_python_dispatcher() if shape_env is not None else nullcontext()
613 with enable_python_dispatcher():
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/passes/ |
| H A D | decomp.py | 69 … with fake_tensor.unset_fake_temporarily(), python_dispatch.enable_python_dispatcher(), fake_mode:
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | pass_base.py | 11 from torch._dispatch.python import enable_python_dispatcher
435 dispatcher_mode = enable_python_dispatcher() # type: ignore[assignment]
|
| H A D | __init__.py | 26 from torch._dispatch.python import enable_python_dispatcher
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | eval_frame.py | 55 from torch._dispatch.python import enable_python_dispatcher
1381 ), enable_python_dispatcher(), ignore_fresh_unbacked:
1538 with unset_fake_temporarily(), enable_python_dispatcher(), fake_mode:
|
| H A D | utils.py | 95 from torch._dispatch.python import enable_python_dispatcher
1784 with tx.fake_mode, enable_python_dispatcher():
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | pass_base.py | 35 from torch._dispatch.python import enable_python_dispatcher
572 dispatcher_mode = enable_python_dispatcher() # type: ignore[assignment]
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | _trace.py | 17 from torch._dispatch.python import enable_python_dispatcher
1433 with enable_python_dispatcher():
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | proxy_tensor.py | 48 from torch._dispatch.python import enable_python_dispatcher
1921 self.python_dispatcher_mode = enable_python_dispatcher()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | pattern_matcher.py | 79 from torch._dispatch.python import enable_python_dispatcher
1830 with enable_python_dispatcher():
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_vmap.py | 1141 with torch._dispatch.python.enable_python_dispatcher():
|