Searched refs:unsafe_split (Results 1 – 14 of 14) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | preprocess_for_onnx.cpp | 97 case aten::unsafe_split: in FuseWithListUnpack()
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | __init__.py | 466 aten.unsafe_split.Tensor,
|
| H A D | decompositions.py | 1443 @register_decomposition(aten.unsafe_split.Tensor)
1444 def unsafe_split(input: Tensor, split_size: int, dim: int = 0) -> Tuple[Tensor, ...]: function
1948 return torch.ops.aten.unsafe_split.Tensor(tensor, split_size, dim)
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | annotated_ops | 250 unsafe_split, view/reshape
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/tests/ |
| H A D | test_replace_ops_passes.py | 294 model = TensorSplitWithSizes(split_size, dim, torch.unsafe_split)
306 count_node(graph_after_passes, exir_ops.edge.aten.unsafe_split.Tensor),
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesViews.cpp | 586 VMAP_SUPPORT2(unsafe_split, Tensor, unsafe_split_batch_rule); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _torch_docs.py | 1968 torch.unsafe_split,
|
| H A D | overrides.py | 1246 torch.unsafe_split: lambda tensor, split_size_or_sections, dim=0: -1,
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset13.py | 124 def unsafe_split( function
|
| H A D | symbolic_opset9.py | 1052 def unsafe_split( function
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 1316 aten::unsafe_split.Tensor_out
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/ |
| H A D | replace_ops.py | 47 exir_ops.edge.aten.unsafe_split.Tensor: exir_ops.edge.aten.split_copy.Tensor,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorShape.cpp | 1046 return self.unsafe_split(split_size, dim); in unsafe_chunk()
2588 std::vector<Tensor> unsafe_split(const Tensor& self, int64_t split_size, int64_t dim) { in unsafe_split() function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_autograd.py | 11911 lambda t: t.unsafe_split(1),
|