diff options
-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 |