| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_pytree.py | 492 def run_test(pytree): argument
520 def run_test(pytree, one_level_leaves): argument
565 def run_test(pytree): argument
599 def run_test(pytree): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/ |
| H A 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
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/ |
| H A 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
|
| /aosp_15_r20/external/executorch/extension/pytree/test/ |
| H A D | test_pytree.cpp | 144 TEST(pytree, FlattenDict) { in TEST() argument
161 TEST(pytree, FlattenNestedDict) { in TEST() argument
|
| H A D | test.py | 224 def run_test(pytree): argument
242 def run_test(pytree): argument
|
| /aosp_15_r20/external/executorch/extension/pytree/ |
| H A D | pybindings.cpp | 20 namespace pytree { namespace
301 const auto& pytree = p.second; in tree_map() local
|
| /aosp_15_r20/external/pytorch/torch/_export/db/examples/ |
| H A D | pytree_flatten.py | 4 from torch.utils import _pytree as pytree unknown
|
| /aosp_15_r20/external/executorch/exir/passes/ |
| H A D | insert_write_back_for_buffers_pass.py | 19 from torch.utils import _pytree as pytree unknown
|
| H A D | quant_fusion_pass.py | 12 from torch.utils import _pytree as pytree unknown
|
| H A D | const_prop_pass.py | 15 from torch.utils import _pytree as pytree unknown
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/optests/ |
| H A D | autograd_registration.py | 6 import torch.utils._pytree as pytree namespace
|
| /aosp_15_r20/external/executorch/devtools/bundled_program/test/ |
| H A D | test_end2end.py | 20 import executorch.extension.pytree as pytree namespace
|
| /aosp_15_r20/external/executorch/exir/capture/ |
| H A D | _unlift.py | 13 from torch.utils import _pytree as pytree unknown
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | memory.py | 15 from torch.utils import _pytree as pytree unknown
|
| H A D | pass_manager.py | 13 import torch.utils._pytree as pytree namespace
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/ |
| H A D | _passes.py | 13 import torch.utils._pytree as pytree namespace
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/backends/ |
| H A D | cudagraphs.py | 7 from torch.utils import _pytree as pytree unknown
|
| /aosp_15_r20/external/executorch/exir/emit/ |
| H A D | _emit_program.py | 24 from torch.utils import _pytree as pytree unknown
|
| /aosp_15_r20/external/pytorch/torch/export/passes/ |
| H A D | __init__.py | 4 import torch.utils._pytree as pytree namespace
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | _per_sample_grad.py | 6 from torch.utils import _pytree as pytree unknown
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/ |
| H A D | common_op_utils.py | 5 from torch.utils import _pytree as pytree unknown
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | compile_utils.py | 9 from torch.utils import _pytree as pytree unknown
|
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | collect_metadata_analysis.py | 18 import torch.utils._pytree as pytree namespace
|
| H A D | dispatch_and_compile_graph.py | 11 import torch.utils._pytree as pytree namespace
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | __init__.py | 5 import torch.utils._pytree as pytree namespace
|