Home
last modified time | relevance | path

Searched refs:nnTensorDesc (Results 1 – 6 of 6) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dhdi_prepared_model_v1_0.cpp104 TensorDesc* nnTensorDesc = nnTensor->GetTensorDesc(); in TransIOTensor() local
105 if (nnTensorDesc == nullptr) { in TransIOTensor()
112 OH_NN_ReturnCode ret = nnTensorDesc->GetName(&tensorName); in TransIOTensor()
121 ret = nnTensorDesc->GetDataType(&dataType); in TransIOTensor()
130 ret = nnTensorDesc->GetFormat(&format); in TransIOTensor()
140 ret = nnTensorDesc->GetShape(&shape, &shapeNum); in TransIOTensor()
H A Dhdi_prepared_model_v2_0.cpp105 TensorDesc* nnTensorDesc = nnTensor->GetTensorDesc(); in TransIOTensor() local
106 if (nnTensorDesc == nullptr) { in TransIOTensor()
113 OH_NN_ReturnCode ret = nnTensorDesc->GetName(&tensorName); in TransIOTensor()
122 ret = nnTensorDesc->GetDataType(&dataType); in TransIOTensor()
131 ret = nnTensorDesc->GetFormat(&format); in TransIOTensor()
141 ret = nnTensorDesc->GetShape(&shape, &shapeNum); in TransIOTensor()
H A Dhdi_prepared_model_v2_1.cpp105 TensorDesc* nnTensorDesc = nnTensor->GetTensorDesc(); in TransIOTensor() local
106 if (nnTensorDesc == nullptr) { in TransIOTensor()
113 OH_NN_ReturnCode ret = nnTensorDesc->GetName(&tensorName); in TransIOTensor()
122 ret = nnTensorDesc->GetDataType(&dataType); in TransIOTensor()
131 ret = nnTensorDesc->GetFormat(&format); in TransIOTensor()
141 ret = nnTensorDesc->GetShape(&shape, &shapeNum); in TransIOTensor()
H A Dinner_model.h39 OH_NN_ReturnCode AddTensorDesc(const NN_TensorDesc* nnTensorDesc);
H A Dinner_model.cpp202 OH_NN_ReturnCode InnerModel::AddTensorDesc(const NN_TensorDesc* nnTensorDesc) in AddTensorDesc() argument
204 if (nnTensorDesc == nullptr) { in AddTensorDesc()
215 OH_NN_ReturnCode returnCode = tensor->BuildFromTensorDesc(nnTensorDesc); in AddTensorDesc()
H A Dnnexecutor.cpp297 TensorDesc* nnTensorDesc = nnTensor->GetTensorDesc(); in RunSync() local
298 if (nnTensorDesc == nullptr) { in RunSync()
302 ret = nnTensorDesc->SetShape(outputsDims[i].data(), outputsDims[i].size()); in RunSync()