diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 4 | ||||
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.hpp | 2 | ||||
-rw-r--r-- | src/armnnTfLiteParser/test/Conv3D.cpp | 2 |
3 files changed, 4 insertions, 4 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 91420ab302..33567953a2 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -684,7 +684,7 @@ TfLiteParserImpl::TfLiteParserImpl(const Optional<ITfLiteParser::TfLiteParserOpt m_ParserFunctions[tflite::BuiltinOperator_CONCATENATION] = &TfLiteParserImpl::ParseConcatenation; m_ParserFunctions[tflite::BuiltinOperator_CONV_2D] = &TfLiteParserImpl::ParseConv2D; // Conv3D support was added in TF 2.5, so for backwards compatibility a hash define is needed. - #if defined(ARMNN_POST_TFLITE_2_3) + #if defined(ARMNN_POST_TFLITE_2_4) m_ParserFunctions[tflite::BuiltinOperator_CONV_3D] = &TfLiteParserImpl::ParseConv3D; #endif m_ParserFunctions[tflite::BuiltinOperator_CUSTOM] = &TfLiteParserImpl::ParseCustomOperator; @@ -1160,7 +1160,7 @@ void TfLiteParserImpl::ParseConv2D(size_t subgraphIndex, size_t operatorIndex) } // Conv3D support was added in TF 2.5, so for backwards compatibility a hash define is needed. -#if defined(ARMNN_POST_TFLITE_2_3) +#if defined(ARMNN_POST_TFLITE_2_4) void TfLiteParserImpl::ParseConv3D(size_t subgraphIndex, size_t operatorIndex) { CHECK_MODEL(m_Model, subgraphIndex, operatorIndex); diff --git a/src/armnnTfLiteParser/TfLiteParser.hpp b/src/armnnTfLiteParser/TfLiteParser.hpp index e742d30521..49744a0484 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -120,7 +120,7 @@ private: void ParseConcatenation(size_t subgraphIndex, size_t operatorIndex); void ParseConv2D(size_t subgraphIndex, size_t operatorIndex); // Conv3D support was added in TF 2.5, so for backwards compatibility a hash define is needed. - #if defined(ARMNN_POST_TFLITE_2_3) + #if defined(ARMNN_POST_TFLITE_2_4) void ParseConv3D(size_t subgraphIndex, size_t operatorIndex); #endif void ParseDepthToSpace(size_t subgraphIndex, size_t operatorIndex); diff --git a/src/armnnTfLiteParser/test/Conv3D.cpp b/src/armnnTfLiteParser/test/Conv3D.cpp index dd55aea211..53332107ff 100644 --- a/src/armnnTfLiteParser/test/Conv3D.cpp +++ b/src/armnnTfLiteParser/test/Conv3D.cpp @@ -7,7 +7,7 @@ #include <sstream> // Conv3D support was added in TF 2.5, so for backwards compatibility a hash define is needed. -#if defined(ARMNN_POST_TFLITE_2_3) +#if defined(ARMNN_POST_TFLITE_2_4) TEST_SUITE("TensorflowLiteParser_Conv3D") { struct SimpleConv3DFixture : public ParserFlatbuffersFixture |