diff options
Diffstat (limited to 'source/use_case')
-rw-r--r-- | source/use_case/asr/src/Wav2LetterModel.cc | 4 | ||||
-rw-r--r-- | source/use_case/img_class/usecase.cmake | 4 | ||||
-rw-r--r-- | source/use_case/kws_asr/src/Wav2LetterModel.cc | 4 |
3 files changed, 6 insertions, 6 deletions
diff --git a/source/use_case/asr/src/Wav2LetterModel.cc b/source/use_case/asr/src/Wav2LetterModel.cc index 6f87be8..a22dc55 100644 --- a/source/use_case/asr/src/Wav2LetterModel.cc +++ b/source/use_case/asr/src/Wav2LetterModel.cc @@ -26,9 +26,9 @@ const tflite::MicroOpResolver& arm::app::Wav2LetterModel::GetOpResolver() bool arm::app::Wav2LetterModel::EnlistOperations() { this->m_opResolver.AddConv2D(); - this->m_opResolver.AddMul(); - this->m_opResolver.AddMaximum(); this->m_opResolver.AddReshape(); + this->m_opResolver.AddLeakyRelu(); + this->m_opResolver.AddSoftmax(); #if defined(ARM_NPU) if (kTfLiteOk == this->m_opResolver.AddEthosU()) { diff --git a/source/use_case/img_class/usecase.cmake b/source/use_case/img_class/usecase.cmake index 63a4c2a..e46de00 100644 --- a/source/use_case/img_class/usecase.cmake +++ b/source/use_case/img_class/usecase.cmake @@ -47,9 +47,9 @@ USER_OPTION(${use_case}_ACTIVATION_BUF_SZ "Activation buffer size for the chosen STRING) if (ETHOS_U_NPU_ENABLED) - set(DEFAULT_MODEL_PATH ${DEFAULT_MODEL_DIR}/mobilenet_v2_1.0_224_quantized_1_default_1_vela_H128.tflite) + set(DEFAULT_MODEL_PATH ${DEFAULT_MODEL_DIR}/mobilenet_v2_1.0_224_INT8_vela_H128.tflite) else() - set(DEFAULT_MODEL_PATH ${DEFAULT_MODEL_DIR}/mobilenet_v2_1.0_224_quantized_1_default_1.tflite) + set(DEFAULT_MODEL_PATH ${DEFAULT_MODEL_DIR}/mobilenet_v2_1.0_224_INT8.tflite) endif() USER_OPTION(${use_case}_MODEL_TFLITE_PATH "NN models file to be used in the evaluation application. Model files must be in tflite format." diff --git a/source/use_case/kws_asr/src/Wav2LetterModel.cc b/source/use_case/kws_asr/src/Wav2LetterModel.cc index 62245b9..affa1a6 100644 --- a/source/use_case/kws_asr/src/Wav2LetterModel.cc +++ b/source/use_case/kws_asr/src/Wav2LetterModel.cc @@ -35,8 +35,8 @@ const tflite::MicroOpResolver& arm::app::Wav2LetterModel::GetOpResolver() bool arm::app::Wav2LetterModel::EnlistOperations() { this->m_opResolver.AddConv2D(); - this->m_opResolver.AddMul(); - this->m_opResolver.AddMaximum(); + this->m_opResolver.AddLeakyRelu(); + this->m_opResolver.AddSoftmax(); this->m_opResolver.AddReshape(); #if defined(ARM_NPU) |