Home
last modified time | relevance | path

Searched refs:pNnExecution (Results 1 – 2 of 2) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_delegate_kernel.cpp137 OH_NNExecutor* pNnExecution {nullptr}; in Invoke() local
138 pNnExecution = m_nnrt->OH_NNExecutor_Construct(m_pNnCompilation); in Invoke()
139 if (pNnExecution == nullptr) { in Invoke()
146 TF_LITE_ENSURE_STATUS(SetInputTensors(context, node, pNnExecution, inputNnTensor)); in Invoke()
149 TF_LITE_ENSURE_STATUS(SetOutputTensors(context, node, pNnExecution)); in Invoke()
154 m_nnrt->OH_NNExecutor_Destroy(&pNnExecution); in Invoke()
155 pNnExecution = nullptr; in Invoke()
323 OH_NNExecutor* pNnExecution, OH_NN_Tensor& nnTensor) in SetInputTensors() argument
326 TF_LITE_ENSURE_EQ(context, pNnExecution != nullptr, true); in SetInputTensors()
360 OH_NNExecutor* pNnExecution) in SetOutputTensors() argument
[all …]
H A Dnnrt_delegate_kernel.h72 …TfLiteStatus SetInputTensors(TfLiteContext* context, TfLiteNode* node, OH_NNExecutor* pNnExecution,
74 …LiteStatus SetOutputTensors(TfLiteContext* context, TfLiteNode* node, OH_NNExecutor* pNnExecution);