diff options
author | Cathal Corbett <cathal.corbett@arm.com> | 2022-05-25 11:21:11 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-08-05 15:52:42 +0100 |
commit | 80b4ef074239c56900ffe27ec5cc47a295deebbb (patch) | |
tree | a555c3fc9ecb25e0cbb15066a943f5baad2a7197 /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | 91b574f57ebf90fbb6445ddc25d215029c330a13 (diff) | |
download | armnn-80b4ef074239c56900ffe27ec5cc47a295deebbb.tar.gz |
Fix '#if defined(ARMNN_POST_TFLITE_2_3)' in TfLiteParser Conv3D test.
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: I5c68b81a67fc2b5a33cf62753351440564bb868e
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 4 |
1 files changed, 2 insertions, 2 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); |