| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | fx.experimental.rst | 56 torch.fx.experimental.proxy_tensor
59 .. currentmodule:: torch.fx.experimental.proxy_tensor
60 .. automodule:: torch.fx.experimental.proxy_tensor
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | compiled_autograd.py | 19 from torch.fx.experimental.proxy_tensor import (
446 proxy_tensor = fetch_object_proxy(self.fx_tracer, t)
447 assert isinstance(proxy_tensor, torch.fx.experimental.proxy_tensor._ProxyTensor)
448 return proxy_tensor.proxy
|
| H A D | _trace_wrapped_higher_order_op.py | 8 from torch.fx.experimental.proxy_tensor import ProxyTorchDispatchMode, track_tensor_tree
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/passes/ |
| H A D | decomp.py | 11 from torch.fx.experimental import proxy_tensor
70 decomposed_module = proxy_tensor.make_fx(
|
| H A D | functionalization.py | 12 from torch.fx.experimental import proxy_tensor
120 graph_module = proxy_tensor.make_fx(
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | utils.py | 11 from torch.fx.experimental.proxy_tensor import make_fx
82 from torch.fx.experimental.proxy_tensor import _CURRENT_MAKE_FX_TRACER
97 from torch.fx.experimental.proxy_tensor import _CURRENT_MAKE_FX_TRACER
|
| H A D | effects.py | 13 from torch.fx.experimental.proxy_tensor import (
242 from torch.fx.experimental.proxy_tensor import (
|
| H A D | run_const_graph.py | 7 from torch.fx.experimental.proxy_tensor import ProxyTorchDispatchMode, track_tensor_tree
|
| H A D | strict_mode.py | 10 from torch.fx.experimental.proxy_tensor import (
|
| H A D | torchbind.py | 12 from torch.fx.experimental.proxy_tensor import ProxyTorchDispatchMode, track_tensor_tree
|
| /aosp_15_r20/external/pytorch/torch/csrc/dynamo/ |
| H A D | compiled_autograd.h | 114 at::Tensor proxy_tensor; member
590 TORCH_INTERNAL_ASSERT(arg.proxy_tensor.defined()); in before()
591 t = arg.proxy_tensor; in before()
603 TORCH_INTERNAL_ASSERT(arg.proxy_tensor.defined()); in before()
604 t = SavedVariable(arg.proxy_tensor, false); in before()
|
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | dispatch_and_compile_graph.py | 18 from torch.fx.experimental.proxy_tensor import make_fx
140 proxy = torch.fx.experimental.proxy_tensor.get_proxy_slot(
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | sym_node.py | 1054 from torch.fx.experimental.proxy_tensor import (
1132 from torch.fx.experimental.proxy_tensor import (
1174 from torch.fx.experimental.proxy_tensor import (
1219 from torch.fx.experimental.proxy_tensor import (
1271 from torch.fx.experimental.proxy_tensor import (
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | python_key.py | 7 from torch.fx.experimental.proxy_tensor import (
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | _safeguard.py | 3 from torch.fx.experimental.proxy_tensor import ProxyTorchDispatchMode
|
| /aosp_15_r20/external/pytorch/torch/_prims/ |
| H A D | executor.py | 6 from torch.fx.experimental.proxy_tensor import make_fx, wrapper_and_args_for_make_fx
|
| /aosp_15_r20/external/executorch/backends/qualcomm/_passes/ |
| H A D | decompose_einsum.py | 9 from torch.fx.experimental.proxy_tensor import make_fx
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_after_aot.py | 12 from torch.fx.experimental.proxy_tensor import make_fx
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/ |
| H A D | fake_tensor_prop.py | 9 from torch.fx.experimental.proxy_tensor import snapshot_fake, py_sym_types
|
| /aosp_15_r20/external/executorch/backends/mediatek/_passes/ |
| H A D | decompose_scaled_dot_product_attention.py | 12 from torch.fx.experimental.proxy_tensor import make_fx
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/optests/ |
| H A D | make_fx.py | 4 from torch.fx.experimental.proxy_tensor import make_fx
|
| /aosp_15_r20/external/executorch/backends/transforms/ |
| H A D | decompose_sdpa.py | 12 from torch.fx.experimental.proxy_tensor import make_fx
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/shape_inference/ |
| H A D | infer_shape.py | 8 from torch.fx.experimental.proxy_tensor import make_fx
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_prims.py | 160 from torch.fx.experimental.proxy_tensor import make_fx
436 from torch.fx.experimental.proxy_tensor import make_fx
|
| /aosp_15_r20/external/pytorch/test/fx/ |
| H A D | test_common_passes.py | 6 from torch.fx.experimental.proxy_tensor import make_fx
|