Searched refs:DataDim (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/ir/ |
| D | meta_tensor_test.cc | 141 ASSERT_EQ(0, tensor.DataDim()); in TEST_F()
163 ASSERT_EQ(0, tensor_ptr->DataDim()); in TEST_F()
278 std::cout << "Dim: " << tensor->DataDim() << std::endl; in TEST_F()
279 ASSERT_EQ(2, tensor->DataDim()); in TEST_F()
300 std::cout << "Dim: " << tensor->DataDim() << std::endl; in TEST_F()
301 ASSERT_EQ(2, tensor->DataDim()); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/ir/ |
| D | tensor_py.cc | 276 …tensor.data_c(), tensor.data().itemsize(), GetPyTypeFormat(tensor.data_type()), tensor.DataDim(), … in GetPyBufferInfo()
597 .def("dim", &Tensor::DataDim, R"mydelimiter( in __anonff91c9900102()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | tensor.h | 224 int DataDim() const { return static_cast<int>(data().ndim()); } in DataDim() function
|
| D | pattern_matcher.h | 618 if ((tensor_ptr->DataSize() > 1) || (tensor_ptr->DataDim() > 0)) {
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_helper.cc | 126 auto origin_ndim = IntToSize(tensor->DataDim()); in ConvertToScalarTensor()
|