| /aosp_15_r20/external/executorch/kernels/aten/ |
| H A D | functions.yaml | 400 - op: unsafe_split.Tensor_out
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | schema_check_mode.py | 177 unsafe_ops = ("aten::_unsafe_view", "aten::unsafe_split")
|
| /aosp_15_r20/external/pytorch/test/edge/ |
| H A D | selected_operators.yaml | 416 aten::unsafe_split.Tensor_out:
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | annotated_ops | 250 unsafe_split, view/reshape
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset13.py | 123 @_onnx_symbolic("aten::unsafe_split")
124 def unsafe_split( function
|
| H A D | symbolic_opset9.py | 281 "unsafe_split",
1051 @_onnx_symbolic("aten::unsafe_split")
1052 def unsafe_split( function
|
| /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/torch/csrc/jit/passes/onnx/ |
| H A D | preprocess_for_onnx.cpp | 97 case aten::unsafe_split: in FuseWithListUnpack()
|
| /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/docs/source/ |
| H A D | conf.py | 1407 "unsafe_split",
1713 "unsafe_split",
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_dtensor_ops.py | 462 xfail("unsafe_split"),
|
| /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/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 1316 aten::unsafe_split.Tensor_out
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | NamedRegistrations.cpp | 501 m.impl("unsafe_split.Tensor", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _tensor_docs.py | 6353 "unsafe_split",
6355 unsafe_split(split_size, dim=0) -> List of Tensors
6357 See :func:`torch.unsafe_split`
|
| H A D | overrides.py | 1246 torch.unsafe_split: lambda tensor, split_size_or_sections, dim=0: -1,
|
| H A D | _torch_docs.py | 1968 torch.unsafe_split,
1970 unsafe_split(tensor, split_size_or_sections, dim=0) -> List of Tensors
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 523 skip("unsafe_split"), # slow: takes 49 sec on A100
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | gen_variable_type.py | 243 "unsafe_split",
|
| H A D | derivatives.yaml | 1565 - name: unsafe_split.Tensor(Tensor self, SymInt split_size, int dim=0) -> Tensor[]
|
| /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
|
| H A D | native_functions.yaml | 5518 - func: unsafe_split.Tensor(Tensor self, SymInt split_size, int dim=0) -> Tensor[]
5523 CompositeExplicitAutograd: unsafe_split
5524 autogen: unsafe_split.Tensor_out
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 2183 "torch.unsafe_split",
|