aboutsummaryrefslogtreecommitdiff
path: root/src/armnnTfLiteParser/TfLiteParser.cpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2022-09-22 10:12:58 +0100
committerTeresaARM <teresa.charlinreyes@arm.com>2022-09-22 10:57:32 +0000
commitbc37a6b83faf92c92570fb3137b5fd549f304b3f (patch)
tree203dd10b33108a27251b3fff555e9b045fe76482 /src/armnnTfLiteParser/TfLiteParser.cpp
parent49ed0df12338b1e99674edeee4200acf8c05750e (diff)
downloadarmnn-bc37a6b83faf92c92570fb3137b5fd549f304b3f.tar.gz
IVGCVSW-7240 Adjoint is Transpose in TFLite. Change in TFLite parser
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: I1bc3d50b8fa6e216d8b6b7e3421d2ff37a21712c
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r--src/armnnTfLiteParser/TfLiteParser.cpp7
1 files changed, 4 insertions, 3 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp
index a26f3e5f04..e036d0ca1c 100644
--- a/src/armnnTfLiteParser/TfLiteParser.cpp
+++ b/src/armnnTfLiteParser/TfLiteParser.cpp
@@ -1586,10 +1586,11 @@ void TfLiteParserImpl::ParseBatchMatMul(size_t subgraphIndex, size_t operatorInd
const auto& operatorPtr = m_Model->subgraphs[subgraphIndex]->operators[operatorIndex];
const auto* options = operatorPtr->builtin_options.AsBatchMatMulOptions();
- BatchMatMulDescriptor descriptor(false,
+ // Adjoint in tensorflow lite performs transpose operation
+ BatchMatMulDescriptor descriptor(options->adj_x,
+ options->adj_y,
false,
- options->adj_x,
- options->adj_y);
+ false);
// Arbitrary DataLayout
IConnectableLayer* layer = m_Network->AddBatchMatMulLayer(descriptor, layerName.c_str());