diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-09-22 10:12:58 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-09-22 10:57:32 +0000 |
commit | bc37a6b83faf92c92570fb3137b5fd549f304b3f (patch) | |
tree | 203dd10b33108a27251b3fff555e9b045fe76482 /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | 49ed0df12338b1e99674edeee4200acf8c05750e (diff) | |
download | armnn-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.cpp | 7 |
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()); |