Searched defs:split_with_sizes (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset13.py | 119 def split_with_sizes(g: jit_utils.GraphContext, self, split_sizes, dim, _outputs=None): function
|
| H A D | symbolic_opset11.py | 638 def split_with_sizes(g: jit_utils.GraphContext, self, split_sizes, dim, _outputs=None): function
|
| H A D | symbolic_opset9.py | 1060 def split_with_sizes(g: jit_utils.GraphContext, self, split_sizes, dim, _outputs=None): function
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 1397 def split_with_sizes( function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 1619 def split_with_sizes(x, sizes, dim=0): function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorShape.cpp | 2621 std::vector<Tensor> split_with_sizes(const Tensor& self, IntArrayRef split_sizes, int64_t dim) { in split_with_sizes() function
|