Searched refs:unsafe_split_with_sizes (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | preprocess_for_onnx.cpp | 98 case aten::unsafe_split_with_sizes: in FuseWithListUnpack()
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | __init__.py | 467 aten.unsafe_split_with_sizes,
|
| H A D | decompositions.py | 1448 @register_decomposition(aten.unsafe_split_with_sizes.default)
1449 def unsafe_split_with_sizes( function
1947 return torch.ops.aten.unsafe_split_with_sizes.default(tensor, split_sizes, dim)
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | annotated_ops | 252 unsafe_split_with_sizes, view/reshape
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset13.py | 131 def unsafe_split_with_sizes( function
|
| H A D | symbolic_opset9.py | 1069 def unsafe_split_with_sizes( function
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 1317 aten::unsafe_split_with_sizes.out
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorShape.cpp | 1044 return self.unsafe_split_with_sizes(split_sizes, dim); in unsafe_chunk()
2643 std::vector<Tensor> unsafe_split_with_sizes(const Tensor& self, IntArrayRef split_sizes, int64_t di… in unsafe_split_with_sizes() function
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | overrides.py | 1247 torch.unsafe_split_with_sizes: lambda tensor, split_size_or_sections, dim=0: -1,
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_autograd.py | 11912 lambda t: t.unsafe_split_with_sizes((1, 1, 1)),
|