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 | |
| 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')
| -rw-r--r-- | delegate/TensorFlowLiteDelegateSupport.md | 2 | ||||
| -rw-r--r-- | delegate/src/FullyConnected.hpp | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/delegate/TensorFlowLiteDelegateSupport.md b/delegate/TensorFlowLiteDelegateSupport.md index a9f548e3b1..b1b39f616b 100644 --- a/delegate/TensorFlowLiteDelegateSupport.md +++ b/delegate/TensorFlowLiteDelegateSupport.md @@ -26,7 +26,7 @@ The Arm NN SDK TensorFlow Lite delegate currently supports the following operato * EXP -* FULLY_CONNECTED +* FULLY_CONNECTED, Supported Fused Activation: RELU , RELU6 , TANH, NONE * GREATER 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 |