Home
last modified time | relevance | path

Searched refs:m_keyToOpFunc (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_op_builder.cpp467 if (!m_keyToOpFunc.count(builtinCode)) { in AddOpFuncParams()
472 OpFuncPtr pfunc = m_keyToOpFunc[builtinCode]; in AddOpFuncParams()
478 m_keyToOpFunc[kTfLiteBuiltinAdd] = &NnrtOpBuilder::AddBasicComputeParams; in MapBuiltinCodeToFunc()
481 m_keyToOpFunc[kTfLiteBuiltinConv2d] = &NnrtOpBuilder::AddConv2DParams; in MapBuiltinCodeToFunc()
486 m_keyToOpFunc[kTfLiteBuiltinMul] = &NnrtOpBuilder::AddBasicComputeParams; in MapBuiltinCodeToFunc()
488 m_keyToOpFunc[kTfLiteBuiltinReshape] = &NnrtOpBuilder::AddReshapeParams; in MapBuiltinCodeToFunc()
489 m_keyToOpFunc[kTfLiteBuiltinSoftmax] = &NnrtOpBuilder::AddSoftmaxParams; in MapBuiltinCodeToFunc()
491 m_keyToOpFunc[kTfLiteBuiltinPack] = &NnrtOpBuilder::AddPackParams; in MapBuiltinCodeToFunc()
492 m_keyToOpFunc[kTfLiteBuiltinPad] = &NnrtOpBuilder::AddPadParams; in MapBuiltinCodeToFunc()
493 m_keyToOpFunc[kTfLiteBuiltinMean] = &NnrtOpBuilder::AddReduceMeanParams; in MapBuiltinCodeToFunc()
[all …]
H A Dnnrt_op_builder.h319 std::map<int32_t, OpFuncPtr> m_keyToOpFunc; variable