Searched defs:indices_or_sections (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _funcs_impl.py | 203 def _split_helper(tensor, indices_or_sections, axis, strict=False): argument
213 def _split_helper_int(tensor, indices_or_sections, axis, strict=False): argument
240 def _split_helper_list(tensor, indices_or_sections, axis): argument
257 def array_split(ary: ArrayLike, indices_or_sections, axis=0): argument
261 def split(ary: ArrayLike, indices_or_sections, axis=0): argument
265 def hsplit(ary: ArrayLike, indices_or_sections): argument
272 def vsplit(ary: ArrayLike, indices_or_sections): argument
278 def dsplit(ary: ArrayLike, indices_or_sections): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| H A D | np_array_ops.py | 1007 def split(ary, indices_or_sections, axis=0): argument
1017 def f(ary, indices_or_sections): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset13.py | 140 g: jit_utils.GraphContext, self, indices_or_sections, dim, _outputs=None argument
|