Home
last modified time | relevance | path

Searched refs:NnrtDelegateKernel (Results 1 – 5 of 5) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_delegate_kernel.h33 class NnrtDelegateKernel {
35 explicit NnrtDelegateKernel(const NnrtApi* nnrt) in NnrtDelegateKernel() function
43 NnrtDelegateKernel() : NnrtDelegateKernel(NnrtImplementation()) {} in NnrtDelegateKernel() function
44 virtual ~NnrtDelegateKernel() in ~NnrtDelegateKernel()
H A Dnnrt_delegate_kernel.cpp55 bool NnrtDelegateKernel::Validate(const int32_t builtinCode) in Validate()
65 TfLiteStatus NnrtDelegateKernel::Init(TfLiteContext* context, const TfLiteDelegateParams* params) in Init()
96 TfLiteStatus NnrtDelegateKernel::Prepare(TfLiteContext* context, TfLiteNode* node) in Prepare()
128 TfLiteStatus NnrtDelegateKernel::Invoke(TfLiteContext* context, TfLiteNode* node) in Invoke()
159 TfLiteStatus NnrtDelegateKernel::Map(const int32_t builtinCode, const NnrtOpMappingArgs& mappingArg… in Map()
178 TfLiteStatus NnrtDelegateKernel::BuildGraph(TfLiteContext* context, const NnrtDelegate::Options& de… in BuildGraph()
228 TfLiteStatus NnrtDelegateKernel::AddOpsAndTensors(TfLiteContext* context, const TfLiteIntArray* inp… in AddOpsAndTensors()
289 TfLiteStatus NnrtDelegateKernel::ConvertTensorTypeToNn(TfLiteContext* context, in ConvertTensorTypeToNn()
322 TfLiteStatus NnrtDelegateKernel::SetInputTensors(TfLiteContext* context, TfLiteNode* node, in SetInputTensors()
359 TfLiteStatus NnrtDelegateKernel::SetOutputTensors(TfLiteContext* context, TfLiteNode* node, in SetOutputTensors()
[all …]
H A Dnnrt_delegate.cpp158 if (NnrtDelegateKernel::Validate(registration->builtin_code)) { in GetSupportedNodes()
184 NnrtDelegateKernel* state = new (std::nothrow) NnrtDelegateKernel(delegateData->nnrt); in GetDelegateKernelRegistration()
201 delete static_cast<NnrtDelegateKernel*>(buffer); in GetDelegateKernelRegistration()
212 NnrtDelegateKernel* state = reinterpret_cast<NnrtDelegateKernel*>(node->user_data); in GetDelegateKernelRegistration()
222 NnrtDelegateKernel* state = reinterpret_cast<NnrtDelegateKernel*>(node->user_data); in GetDelegateKernelRegistration()
H A Dnnrt_delegate.h32 class NnrtDelegateKernel; variable
36 using tflite::delegate::nnrt::NnrtDelegateKernel;
/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/
H A DREADME_zh.md52 …- 调用TFLiteRegistration注册NnrtDelegate，并初始化init，prepare，invoke成员函数指针，指向NnrtDelegateKernel的Init，Prepa…
54 6. 用户调用AllocateTensors，完成tensors内存分配和图编译。其中，支持在NNRtDelegate上运行的node会调用NnrtDelegateKernel的prepare接口完…
72 │   ├── nnrt_delegate_kernel.cpp # NnrtDelegateKernel源文件，将TensorFlow Lite模型中的operator…
73 │   ├── nnrt_delegate_kernel.h # NnrtDelegateKernel头文件
107 if (NnrtDelegateKernel::Validate(registration->builtin_code)) {
116 …- 注册的Delegate kernel，初始化TfLiteRegistration的init，prepare，invoke成员函数指针，指向NnrtDelegateKernel的Init，Pre…
125 NnrtDelegateKernel* state = new (std::nothrow) NnrtDelegateKernel(delegateData->nnrt);
127 TFLITE_LOG_PROD(TFLITE_LOG_ERROR, "Failed to create NnrtDelegateKernel instance.");
133 TFLITE_LOG_PROD(TFLITE_LOG_ERROR, "Failed to init NnrtDelegateKernel.");
146 NnrtDelegateKernel* state = reinterpret_cast<NnrtDelegateKernel*>(node->user_data);
[all …]