Searched refs:split_with_sizes_copy (Results 1 – 18 of 18) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/ |
| H A D | _fsdp_collectives.py | 98 def split_with_sizes_copy( function
104 torch.split_with_sizes_copy(
275 torch.ops.fsdp.split_with_sizes_copy(
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | comms.py | 516 torch.ops.fsdp.split_with_sizes_copy.default,
545 cur_snode.node, torch.ops.fsdp.split_with_sizes_copy.default
|
| /aosp_15_r20/external/executorch/backends/arm/_passes/ |
| H A D | convert_split_to_slice.py | 22 exir_ops.edge.aten.split_with_sizes_copy.default,
|
| /aosp_15_r20/external/executorch/backends/arm/operator_support/ |
| H A D | tosa_supported_operators.py | 87 exir_ops.edge.aten.split_with_sizes_copy.default,
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Split.cpp | 138 aten.split_with_sizes_copy.default, split_with_sizes_copy_default);
|
| /aosp_15_r20/external/executorch/backends/vulkan/partitioner/ |
| H A D | supported_ops.py | 122 exir_ops.edge.aten.split_with_sizes_copy.default,
|
| /aosp_15_r20/external/executorch/exir/tests/ |
| H A D | test_delegate.py | 297 node.target == exir_ops.edge.aten.split_with_sizes_copy.default
|
| /aosp_15_r20/external/executorch/backends/arm/ |
| H A D | tosa_quant_utils.py | 32 exir_ops.edge.aten.split_with_sizes_copy.default,
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | __init__.py | 438 aten.split_with_sizes_copy,
|
| H A D | decompositions.py | 1425 [aten.split_with_sizes_copy.default, aten.split_with_sizes_copy.out]
1427 def split_with_sizes_copy( function
|
| /aosp_15_r20/external/executorch/backends/vulkan/ |
| H A D | op_registry.py | 495 exir_ops.edge.aten.split_with_sizes_copy.default,
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_compile.py | 206 torch.ops.fsdp.split_with_sizes_copy.default,
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ |
| H A D | _tensor_ops.py | 731 aten.split_with_sizes_copy.default,
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _torch_docs.py | 13871 torch.split_with_sizes_copy,
|
| H A D | overrides.py | 1277 torch.split_with_sizes_copy: lambda self, split_sizes, dim=0: -1,
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/tests/ |
| H A D | test_replace_ops_passes.py | 301 graph_after_passes, exir_ops.edge.aten.split_with_sizes_copy.default
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_pattern_matcher.py | 1445 torch.ops.fsdp.split_with_sizes_copy,
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_torch.py | 9432 torch.split_with_sizes_copy(x, split_sizes, dim=dim, out=out)
9447 torch.split_with_sizes_copy(x, split_sizes, dim=dim, out=out)
|