Home
last modified time | relevance | path

Searched defs:shape_num (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/lite/src/c_api/
Dtensor_c.cc22 …e OH_AI_TensorCreate(const char *name, OH_AI_DataType type, const int64_t *shape, size_t shape_num, in OH_AI_TensorCreate()
110 void OH_AI_TensorSetShape(OH_AI_TensorHandle tensor, const int64_t *shape, size_t shape_num) { in OH_AI_TensorSetShape()
123 const int64_t *OH_AI_TensorGetShape(const OH_AI_TensorHandle tensor, size_t *shape_num) { in OH_AI_TensorGetShape()
Dmodel_c.cc109 size_t shape_num = shapes.size(); in Resize() local
/third_party/mindspore/include/sdk_api/
Dmodel.h75 size_t shape_num; member
/third_party/mindspore/include/c_api/
Dmodel_c.h36 size_t shape_num; member
/third_party/mindspore/mindspore/lite/test/ut/src/api/
Dtensor_c_test.cc26 constexpr size_t shape_num = 2; in TEST_F() local
/third_party/mindspore/mindspore/lite/test/ohos_test/src/
Dtensor_test.cc35 constexpr size_t shape_num = 2; variable
/third_party/mindspore/mindspore/lite/src/
Dtensor.h289 inline Tensor::Category TensorCategory(const int node_type, const size_t shape_num, const TypeId da… in TensorCategory()
302 auto shape_num = tensor->dims() == nullptr ? 0 : tensor->dims()->size(); in TensorCategory() local
/third_party/mindspore/mindspore/lite/tools/benchmark/
Dbenchmark_c_api.cc304 size_t shape_num; in CompareDataGetTotalBiasAndSize() local