Searched refs:m_nnModel (Results 1 – 4 of 4) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/ |
| H A D | nnrt_delegate_kernel.h | 40 m_nnModel(nullptr),
46 m_nnrt->OH_NNModel_Destroy(&m_nnModel); in ~NnrtDelegateKernel()
91 OH_NNModel* m_nnModel; variable
|
| H A D | nnrt_delegate_kernel.cpp | 82 if (m_nnModel == nullptr) { in Init()
83 m_nnModel = m_nnrt->OH_NNModel_Construct(); in Init()
84 if (m_nnModel == nullptr) { in Init()
111 m_pNnCompilation = m_nnrt->OH_NNCompilation_Construct(m_nnModel); in Prepare()
221 …RETURN_TFLITE_ERROR_IF_NN_ERROR(m_nnrt->OH_NNModel_SpecifyInputsAndOutputs(m_nnModel, &inputIndice… in BuildGraph()
224 RETURN_TFLITE_ERROR_IF_NN_ERROR(m_nnrt->OH_NNModel_Finish(m_nnModel), "finalizing the model"); in BuildGraph()
235 .nnModel = m_nnModel, in AddOpsAndTensors()
|
| H A D | nnrt_op_builder.h | 136 m_nnrt->OH_NNModel_AddTensor(m_nnModel, &tensor), "adding nnTensor"); in AddScalarTensor()
140 m_nnrt->OH_NNModel_SetTensorData(m_nnModel, nnIndex, &value, sizeof(value)), in AddScalarTensor()
175 m_nnrt->OH_NNModel_AddTensor(m_nnModel, &tensor), "adding nnTensor"); in AddVectorTensor()
178 … m_nnrt->OH_NNModel_SetTensorData(m_nnModel, nnIndex, values, sizeof(*(values)) * numValues), in AddVectorTensor()
302 OH_NNModel* const m_nnModel; variable
|
| H A D | nnrt_op_builder.cpp | 39 m_nnModel(opBuilderArgs.nnModel), in NnrtOpBuilder()
316 RETURN_TFLITE_ERROR_IF_NN_ERROR(m_nnrt->OH_NNModel_AddOperation(m_nnModel, in FinalizeAddOperation()
378 int32_t nnRet = m_nnrt->OH_NNModel_AddTensor(m_nnModel, &nnTensor); in AddTensor()
387 nnRet = m_nnrt->OH_NNModel_SetTensorData(m_nnModel, nnTensorIndex, in AddTensor()
|