| /external/pytorch/torch/distributed/tensor/experimental/ |
| D | _func_map.py | 12 from torch.utils import _cxx_pytree as pytree unknown
14 from torch.utils import _pytree as pytree # type: ignore[no-redef] unknown
|
| /external/pytorch/torch/_export/db/examples/ |
| D | pytree_flatten.py | 4 from torch.utils import _pytree as pytree unknown
|
| /external/executorch/exir/passes/ |
| D | insert_write_back_for_buffers_pass.py | 19 from torch.utils import _pytree as pytree unknown
|
| D | quant_fusion_pass.py | 12 from torch.utils import _pytree as pytree unknown
|
| D | const_prop_pass.py | 15 from torch.utils import _pytree as pytree unknown
|
| /external/pytorch/torch/testing/_internal/optests/ |
| D | autograd_registration.py | 6 import torch.utils._pytree as pytree namespace
|
| /external/executorch/devtools/bundled_program/test/ |
| D | test_end2end.py | 20 import executorch.extension.pytree as pytree namespace
|
| /external/executorch/exir/capture/ |
| D | _unlift.py | 13 from torch.utils import _pytree as pytree unknown
|
| /external/executorch/exir/ |
| D | memory.py | 15 from torch.utils import _pytree as pytree unknown
|
| D | pass_manager.py | 13 import torch.utils._pytree as pytree namespace
|
| /external/pytorch/torch/distributed/tensor/ |
| D | _dispatch.py | 36 from torch.utils import _cxx_pytree as pytree unknown
38 from torch.utils import _pytree as pytree # type: ignore[no-redef] unknown
|
| /external/pytorch/torch/fx/passes/backends/ |
| D | cudagraphs.py | 7 from torch.utils import _pytree as pytree unknown
|
| /external/executorch/backends/cadence/aot/ |
| D | _passes.py | 13 import torch.utils._pytree as pytree namespace
|
| /external/pytorch/torch/export/passes/ |
| D | __init__.py | 4 import torch.utils._pytree as pytree namespace
|
| /external/executorch/exir/emit/ |
| D | _emit_program.py | 24 from torch.utils import _pytree as pytree unknown
|
| /external/pytorch/torch/nn/utils/ |
| D | _per_sample_grad.py | 6 from torch.utils import _pytree as pytree unknown
|
| /external/pytorch/torch/distributed/_shard/ |
| D | common_op_utils.py | 5 from torch.utils import _pytree as pytree unknown
|
| /external/pytorch/torch/_functorch/ |
| D | compile_utils.py | 9 from torch.utils import _pytree as pytree unknown
|
| /external/pytorch/torch/_functorch/_aot_autograd/ |
| D | collect_metadata_analysis.py | 18 import torch.utils._pytree as pytree namespace
|
| D | dispatch_and_compile_graph.py | 11 import torch.utils._pytree as pytree namespace
|
| /external/pytorch/torch/_inductor/ |
| D | __init__.py | 5 import torch.utils._pytree as pytree namespace
|
| /external/executorch/extension/pytree/test/ |
| D | test_pytree.cpp | 144 TEST(pytree, FlattenDict) { in TEST() argument
161 TEST(pytree, FlattenNestedDict) { in TEST() argument
|
| D | test.py | 224 def run_test(pytree): argument
242 def run_test(pytree): argument
|
| /external/pytorch/torch/onnx/_internal/fx/passes/ |
| D | functionalization.py | 15 from torch.utils import _pytree as pytree unknown
|
| /external/pytorch/torch/_higher_order_ops/ |
| D | run_const_graph.py | 8 from torch.utils import _pytree as pytree unknown
|