diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-11-25 18:34:51 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-11-26 16:27:23 +0000 |
commit | 1c717648a51af9058db90301fba3451845674ee2 (patch) | |
tree | 1f438459ad6bc0d182175767fd9e82a8dfa1d523 /delegate/src | |
parent | fc55a19c73240236d30482a6189f438ffb2e2cc4 (diff) | |
download | armnn-1c717648a51af9058db90301fba3451845674ee2.tar.gz |
Bug fix TfLiteDelegate: wrong operator name in FullyConnected activation
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ic52b27d462f498314050576b5195f759f5f00c63
Diffstat (limited to 'delegate/src')
-rw-r--r-- | delegate/src/FullyConnected.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/delegate/src/FullyConnected.hpp b/delegate/src/FullyConnected.hpp index 53251f7c55..0a82286479 100644 --- a/delegate/src/FullyConnected.hpp +++ b/delegate/src/FullyConnected.hpp @@ -223,7 +223,7 @@ TfLiteStatus VisitFullyConnectedOperator(DelegateData& delegateData, Connect(layer, tfLiteNode, delegateData); } - auto* tfLiteNodeParameters = reinterpret_cast<TfLiteAddParams*>(tfLiteNode->builtin_data); + auto* tfLiteNodeParameters = reinterpret_cast<TfLiteFullyConnectedParams*>(tfLiteNode->builtin_data); if (!tfLiteNodeParameters) { // No Activation |