Searched refs:NnrtOpBuilder (Results 1 – 4 of 4) sorted by relevance
35 NnrtOpBuilder::NnrtOpBuilder(const NnrtApi* nnrt, NnrtOpBuilderArgs& opBuilderArgs) in NnrtOpBuilder() function in tflite::delegate::nnrt::NnrtOpBuilder476 TfLiteStatus NnrtOpBuilder::MapBuiltinCodeToFunc() in MapBuiltinCodeToFunc()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 …]
36 class NnrtOpBuilder; variable41 NnrtOpBuilder* builder {nullptr};57 class NnrtOpBuilder {59 NnrtOpBuilder(const NnrtApi* nnrt, NnrtOpBuilderArgs& opBuilderArgs);60 ~NnrtOpBuilder() = default;318 …using OpFuncPtr = TfLiteStatus(NnrtOpBuilder::*)(const NnrtOpMappingArgs& mappingArgs, int32_t bui…
240 NnrtOpBuilder builder(m_nnrt, opBuilderArgs); in AddOpsAndTensors()
75 │ ├── nnrt_op_builder.cpp # NnrtOpBuilder源文件，给每个operators设置输入输出tensor和operatio…76 │ ├── nnrt_op_builder.h # NnrtOpBuilder头文件