Home
last modified time | relevance | path

Searched refs:m_nnrt (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 Dnnrt_delegate_kernel.cpp81 TF_LITE_ENSURE_STATUS(tflite::GetTargetDevice(context, params->delegate, m_nnrt, m_nnrtDevice)); in Init()
83 m_nnModel = m_nnrt->OH_NNModel_Construct(); in Init()
111 m_pNnCompilation = m_nnrt->OH_NNCompilation_Construct(m_nnModel); in Prepare()
121 RETURN_TFLITE_ERROR_IF_NN_ERROR_FOR_COMPILE(m_nnrt->OH_NNCompilation_Build(m_pNnCompilation), in Prepare()
138 pNnExecution = m_nnrt->OH_NNExecutor_Construct(m_pNnCompilation); in Invoke()
152 RETURN_TFLITE_ERROR_IF_NN_ERROR(m_nnrt->OH_NNExecutor_Run(pNnExecution), "running computation"); in Invoke()
154 m_nnrt->OH_NNExecutor_Destroy(&pNnExecution); in Invoke()
224 RETURN_TFLITE_ERROR_IF_NN_ERROR(m_nnrt->OH_NNModel_Finish(m_nnModel), "finalizing the model"); in BuildGraph()
240 NnrtOpBuilder builder(m_nnrt, opBuilderArgs); in AddOpsAndTensors()
400 m_nnrt->OH_NNCompilation_SetPerformanceMode(m_pNnCompilation, performance), in SetNnOptions()
[all …]
H A Dnnrt_delegate_kernel.h39 m_nnrt(nnrt),
46 m_nnrt->OH_NNModel_Destroy(&m_nnModel); in ~NnrtDelegateKernel()
47 m_nnrt->OH_NNCompilation_Destroy(&m_pNnCompilation); in ~NnrtDelegateKernel()
48 m_nnrt = nullptr; in ~NnrtDelegateKernel()
88 const NnrtApi* m_nnrt; variable
H A Dnnrt_op_builder.h136 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()
290 const NnrtApi* const m_nnrt;
H A Dnnrt_op_builder.cpp36 : m_nnrt(nnrt), 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()