IVGCVSW-2624 Support static quantization of FullyConnected

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

1 files changed, 11 insertions, 2 deletions

diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp
index dcaccd4ac7..53b3edc3fd 100644
--- a/src/armnn/QuantizerVisitor.hpp
+++ b/src/armnn/QuantizerVisitor.hpp
@@ -41,8 +41,17 @@ public:
                                       const ConstTensor& beta,
                                       const ConstTensor& gamma,
                                       const char* name = nullptr) override;
-
-    /// Extract the quantized network
+    void VisitFullyConnectedLayer(const IConnectableLayer *layer,
+                                  const FullyConnectedDescriptor&,
+                                  const ConstTensor&,
+                                  const char *name = nullptr)  override;
+    void VisitFullyConnectedLayer(const IConnectableLayer *layer,
+                                  const FullyConnectedDescriptor&,
+                                  const ConstTensor&,
+                                  const ConstTensor&,
+                                  const char *name = nullptr)  override;
+
+    // Extract the quantized network
     INetworkPtr RetrieveFinalNetwork() { return std::move(m_QuantizedNetwork); }
 
 private: