diff options
author | Jan Eilers <jan.eilers@arm.com> | 2020-07-28 14:00:06 +0100 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2020-07-28 16:43:36 +0100 |
commit | 2f746b3f346d3efa9071fc53592652425869d6b3 (patch) | |
tree | 9985280d6b510733b58fc76c729ad0f8679a287e /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | 8d23636d188838ec13c193949a1a5c7af1c3d8d8 (diff) | |
download | armnn-2f746b3f346d3efa9071fc53592652425869d6b3.tar.gz |
Github#433 Add HardSwish support to TfLiteParser
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: Ic476f8d80bba080ab459db9e6a59cbafd307d129
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 69430134df..1a4449302e 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -538,6 +538,7 @@ TfLiteParser::TfLiteParser(const Optional<ITfLiteParser::TfLiteParserOptions>& o m_ParserFunctions[tflite::BuiltinOperator_DEQUANTIZE] = &TfLiteParser::ParseDequantize; m_ParserFunctions[tflite::BuiltinOperator_EXP] = &TfLiteParser::ParseExp; m_ParserFunctions[tflite::BuiltinOperator_FULLY_CONNECTED] = &TfLiteParser::ParseFullyConnected; + m_ParserFunctions[tflite::BuiltinOperator_HARD_SWISH] = &TfLiteParser::ParseHardSwish; m_ParserFunctions[tflite::BuiltinOperator_LEAKY_RELU] = &TfLiteParser::ParseLeakyRelu; m_ParserFunctions[tflite::BuiltinOperator_LOGISTIC] = &TfLiteParser::ParseLogistic; m_ParserFunctions[tflite::BuiltinOperator_L2_NORMALIZATION] = &TfLiteParser::ParseL2Normalization; @@ -1992,7 +1993,7 @@ void TfLiteParser::ParseRelu6(size_t subgraphIndex, size_t operatorIndex) void TfLiteParser::ParseLeakyRelu(size_t subgraphIndex, size_t operatorIndex) { - ParseActivation(subgraphIndex,operatorIndex, ActivationFunction::LeakyReLu); + ParseActivation(subgraphIndex, operatorIndex, ActivationFunction::LeakyReLu); } void TfLiteParser::ParseLogistic(size_t subgraphIndex, size_t operatorIndex) @@ -2005,6 +2006,10 @@ void TfLiteParser::ParseTanH(size_t subgraphIndex, size_t operatorIndex) ParseActivation(subgraphIndex,operatorIndex,ActivationFunction::TanH); } +void TfLiteParser::ParseHardSwish(size_t subgraphIndex, size_t operatorIndex) +{ + ParseActivation(subgraphIndex, operatorIndex, ActivationFunction::HardSwish); +} void TfLiteParser::ParseActivation(size_t subgraphIndex, size_t operatorIndex, ActivationFunction activationType) { @@ -2055,6 +2060,9 @@ void TfLiteParser::ParseActivation(size_t subgraphIndex, size_t operatorIndex, A activationDesc.m_A = options->alpha; break; } + case ActivationFunction::HardSwish: + layerName += str(boost::format("HARDSWISH:%1%:%2%") % subgraphIndex % operatorIndex); + break; default: { throw ParseException( |