IVGCVSW-2624 Support static quantization of FullyConnected

Change-Id: Ib18085e7e4b2e9d55e29b14122410732e3bdd088
Signed-off-by: FrancisMurtagh <francis.murtagh@arm.com>

1 files changed, 24 insertions, 0 deletions

diff --git a/src/armnn/StaticRangeVisitor.cpp b/src/armnn/StaticRangeVisitor.cpp
index 258d499279..eac434eecb 100644
--- a/src/armnn/StaticRangeVisitor.cpp
+++ b/src/armnn/StaticRangeVisitor.cpp
@@ -87,4 +87,28 @@ void StaticRangeVisitor::VisitActivationLayer(const IConnectableLayer* layer,
     }
 }
 
+void StaticRangeVisitor::VisitFullyConnectedLayer(const armnn::IConnectableLayer *layer,
+                                                  const armnn::FullyConnectedDescriptor& desc,
+                                                  const ConstTensor& weights,
+                                                  const char *name)
+{
+    boost::ignore_unused(desc);
+    boost::ignore_unused(weights);
+    boost::ignore_unused(name);
+    SetRange(layer, 0, -15.0f, 15.0f);
+}
+
+void StaticRangeVisitor::VisitFullyConnectedLayer(const armnn::IConnectableLayer *layer,
+                                                  const armnn::FullyConnectedDescriptor& desc,
+                                                  const ConstTensor& weights,
+                                                  const ConstTensor& bias,
+                                                  const char *name)
+{
+    boost::ignore_unused(desc);
+    boost::ignore_unused(weights);
+    boost::ignore_unused(bias);
+    boost::ignore_unused(name);
+    SetRange(layer, 0, -15.0f, 15.0f);
+}
+
 } //namespace armnn