Searched refs:InsertDim (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | tensorshape_test.cc | 90 t3 = t3.InsertDim(0, 1); // 1, 4, 5, 6 in TEST_F()
91 t3 = t3.InsertDim(2, 2); // 1, 4, 2, 5, 6 in TEST_F()
92 t3 = t3.InsertDim(4, 3); // 1, 4, 2, 5, 3, 6 in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/ |
| D | tensor_shape.cc | 119 TensorShape TensorShape::InsertDim(dsize_t axis, dsize_t dim) const { in InsertDim() function in mindspore::dataset::TensorShape
199 return InsertDim(0, dim); in PrependDim()
|
| D | tensor_shape.h | 98 TensorShape InsertDim(dsize_t axis, dsize_t dim) const;
|
| D | tensor.cc | 621 shape_ = shape_.InsertDim(axis, 1); in ExpandDim()
|