Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_op_builder.h101 return AddTensor(tensorIndex, builtinCode, m_augmentedInputs, tensorFlags);
106 return AddTensor(tensorIndex, builtinCode, m_augmentedOutputs, tensorFlags);
119 TfLiteStatus AddOpFuncParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
187 int32_t builtinCode, OH_NN_TensorType nnTensorType) in AddActivateParamsInOperator() argument
202 builtinParams->activation, builtinCode); in AddActivateParamsInOperator()
244 TfLiteStatus AddConv2DParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
248 TfLiteStatus AddSoftmaxParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
250 TfLiteStatus AddQuantizeParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
252 TfLiteStatus AddPackParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
254 TfLiteStatus AddPadParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
[all …]
H A Dnnrt_op_builder.cpp97 if (builtinCode == kTfLiteBuiltinAdd) { in AddBasicComputeParams()
101 } else if (builtinCode == kTfLiteBuiltinMul) { in AddBasicComputeParams()
105 } else if (builtinCode == kTfLiteBuiltinSub) { in AddBasicComputeParams()
127 … TF_LITE_ENSURE_STATUS(AddPadParamsInOperator<TfLitePoolParams>(mappingArgs, builtin, builtinCode, in AddAvgPoolingParams()
145 … TF_LITE_ENSURE_STATUS(AddPadParamsInOperator<TfLitePoolParams>(mappingArgs, builtin, builtinCode, in AddMaxPoolingParams()
206 TfLiteStatus NnrtOpBuilder::AddPackParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode) in AddPackParams() argument
215 TfLiteStatus NnrtOpBuilder::AddPadParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode) in AddPadParams() argument
342 TF_LITE_ENSURE_STATUS(AddTensor(tensorIndex, builtinCode, tensorFlags, nnTensorIndex)); in AddTensor()
467 if (!m_keyToOpFunc.count(builtinCode)) { in AddOpFuncParams()
472 OpFuncPtr pfunc = m_keyToOpFunc[builtinCode]; in AddOpFuncParams()
[all …]
H A Dnnrt_delegate_kernel.cpp55 bool NnrtDelegateKernel::Validate(const int32_t builtinCode) in Validate() argument
57 if (TFLITE_TYPE_TO_NNRT_TYPE.count(builtinCode) && in Validate()
58 TFLITE_TYPE_TO_NNRT_TYPE.at(builtinCode) != OH_NN_UNSUPPORT_OPS) { in Validate()
159 TfLiteStatus NnrtDelegateKernel::Map(const int32_t builtinCode, const NnrtOpMappingArgs& mappingArg… in Map() argument
162 if (TFLITE_TYPE_TO_NNRT_TYPE.count(builtinCode) == 0) { in Map()
164 … "[NNRT-DELEGATE_KERNEL] Not support current TF-Lite Operator, builtCode: %d.", builtinCode); in Map()
168 TfLiteStatus retValue = mappingArgs.builder->AddOpFuncParams(mappingArgs, builtinCode); in Map()
173 nnOpType = TFLITE_TYPE_TO_NNRT_TYPE.at(builtinCode); in Map()
H A Dnnrt_delegate_kernel.h52 static bool Validate(const int32_t builtinCode);
65 …TfLiteStatus Map(int32_t builtinCode, const NnrtOpMappingArgs& mappingArgs, int32_t& nnOpType) con…
H A Dnnrt_utils.cpp64 bool IsScalarInputSupported(int32_t builtinCode) in IsScalarInputSupported() argument
66 switch (builtinCode) { in IsScalarInputSupported()
H A Dnnrt_utils.h76 extern bool IsScalarInputSupported(int32_t builtinCode);