commit 1c717648a51af9058db90301fba3451845674ee2
author Teresa Charlin <teresa.charlinreyes@arm.com> Wed Nov 25 18:34:51 2020 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Thu Nov 26 16:27:23 2020 +0000
tree 1f438459ad6bc0d182175767fd9e82a8dfa1d523
parent fc55a19c73240236d30482a6189f438ffb2e2cc4

Bug fix TfLiteDelegate: wrong operator name in FullyConnected activation

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ic52b27d462f498314050576b5195f759f5f00c63

delegate/TensorFlowLiteDelegateSupport.md

diff --git a/delegate/TensorFlowLiteDelegateSupport.md b/delegate/TensorFlowLiteDelegateSupport.md
index a9f548e..b1b39f6 100644
--- a/delegate/TensorFlowLiteDelegateSupport.md
+++ b/delegate/TensorFlowLiteDelegateSupport.md
@@ -26,7 +26,7 @@
 
 * EXP
 
-* FULLY_CONNECTED
+* FULLY_CONNECTED, Supported Fused Activation: RELU , RELU6 , TANH, NONE
 
 * GREATER
 

delegate/src/FullyConnected.hpp

diff --git a/delegate/src/FullyConnected.hpp b/delegate/src/FullyConnected.hpp
index 53251f7..0a82286 100644
--- a/delegate/src/FullyConnected.hpp
+++ b/delegate/src/FullyConnected.hpp
@@ -223,7 +223,7 @@
         Connect(layer, tfLiteNode, delegateData);
     }
 
-    auto* tfLiteNodeParameters = reinterpret_cast<TfLiteAddParams*>(tfLiteNode->builtin_data);
+    auto* tfLiteNodeParameters = reinterpret_cast<TfLiteFullyConnectedParams*>(tfLiteNode->builtin_data);
     if (!tfLiteNodeParameters)
     {
         // No Activation