diff options
author | Samuel Yap <samuel.yap@arm.com> | 2022-08-24 17:04:34 +0100 |
---|---|---|
committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-08-31 10:21:01 +0100 |
commit | e7cd8f9d9b2b366b49207c519f5ec5fd0b41c382 (patch) | |
tree | 546406ab4199637a4374d859d0a9ad328a63c97d /src/armnnTfLiteParser/TfLiteParser.hpp | |
parent | b9e6b5c3b96792f40201315c831db0aa257f286c (diff) | |
download | armnn-e7cd8f9d9b2b366b49207c519f5ec5fd0b41c382.tar.gz |
IVGCVSW-6497: BatchMatMul TfLite Parser
* Added armnnTfLiteParser for BatchMatMul
* Added unit testing for parser
* Updated CMakeLists
Signed-off-by: Samuel Yap <samuel.yap@arm.com>
Change-Id: If6842aaf7cf08f688093b714e2ecea6e8cd87161
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.hpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.hpp b/src/armnnTfLiteParser/TfLiteParser.hpp index 49744a0484..f8ddc55649 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -114,6 +114,7 @@ private: void ParseArgMin(size_t subgraphIndex, size_t operatorIndex); void ParseArgMax(size_t subgraphIndex, size_t operatorIndex); void ParseAveragePool2D(size_t subgraphIndex, size_t operatorIndex); + void ParseBatchMatMul(size_t subgraphIndex, size_t operatorIndex); void ParseBatchToSpaceND(size_t subgraphIndex, size_t operatorIndex); void ParseCast(size_t subgraphIndex, size_t operatorIndex); void ParseComparison(size_t subgraphIndex, size_t operatorIndex, armnn::ComparisonOperation comparisonOperation); |