Searched defs:split_size_or_sections (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/st/tensor/ |
| D | test_split.py | 248 def construct(self, x, split_size_or_sections, axis=0): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_split.py | 349 def construct(self, x, split_size_or_sections, axis=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/mint/ |
| D | __init__.py | 731 def split(tensor, split_size_or_sections, dim=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | array_func.py | 4755 def _split_int(x, split_size_or_sections, axis): argument
4780 def _split_sub_tensors(x, split_size_or_sections, axis): argument
4801 def split(tensor, split_size_or_sections, axis=0): argument
4868 def split_ext(tensor, split_size_or_sections, axis=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.cc | 113 Operator CreateSplitOp(int64_t split_size_or_sections, int64_t axis, int64_t index) { in CreateSplitOp()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parse/ |
| D | standard_method.py | 3651 def split(tensor, split_size_or_sections, axis=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | tensor.py | 3743 def split(self, split_size_or_sections, axis=0): argument
|