From eb5f810ad985a8cca831b46d94ee73db5a305c9f Mon Sep 17 00:00:00 2001 From: Matthew Sloyan Date: Tue, 5 Oct 2021 17:31:42 +0100 Subject: IVGCVSW-6165 Add Support for Conv3d to TfLiteParser * Added CONV_3D operator support. * Fixed TfLiteParser test issue where QAsymmU8 tests were incorrectly using the Boolean comparison function. Signed-off-by: Matthew Sloyan Change-Id: I9db03b54b6de63f918717af80c5ac5310fe26183 --- src/armnnTfLiteParser/TfLiteParser.hpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/armnnTfLiteParser/TfLiteParser.hpp') diff --git a/src/armnnTfLiteParser/TfLiteParser.hpp b/src/armnnTfLiteParser/TfLiteParser.hpp index 5645503ca3..8eb529963b 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -113,6 +113,7 @@ private: void ParseComparison(size_t subgraphIndex, size_t operatorIndex, armnn::ComparisonOperation comparisonOperation); void ParseConcatenation(size_t subgraphIndex, size_t operatorIndex); void ParseConv2D(size_t subgraphIndex, size_t operatorIndex); + void ParseConv3D(size_t subgraphIndex, size_t operatorIndex); void ParseDepthToSpace(size_t subgraphIndex, size_t operatorIndex); void ParseDepthwiseConv2D(size_t subgraphIndex, size_t operatorIndex); void ParseDequantize(size_t subgraphIndex, size_t operatorIndex); -- cgit v1.2.1