IVGCVSW-2622 Add static quantization of 2DConvolution

Change-Id: If7985a54eba97f7c61413e0804879e4afbf65c4d
Signed-off-by: Jim Flynn <jim.flynn@arm.com>

1 files changed, 21 insertions, 0 deletions

diff --git a/src/armnn/StaticRangeVisitor.cpp b/src/armnn/StaticRangeVisitor.cpp
index eac434eecb..fa95938b37 100644
--- a/src/armnn/StaticRangeVisitor.cpp
+++ b/src/armnn/StaticRangeVisitor.cpp
@@ -59,6 +59,27 @@ void StaticRangeVisitor::VisitBatchNormalizationLayer(const IConnectableLayer* l
     SetRange(layer, 0, -15.0f, 15.0f);
 }
 
+void StaticRangeVisitor::VisitConvolution2dLayer(const IConnectableLayer* layer,
+                                                 const Convolution2dDescriptor& convolution2dDescriptor,
+                                                 const ConstTensor& weights,
+                                                 const char* name)
+{
+    boost::ignore_unused(convolution2dDescriptor);
+    boost::ignore_unused(weights);
+    boost::ignore_unused(name);
+    SetRange(layer, 0, -15.0f, 15.0f);
+}
+
+void StaticRangeVisitor::VisitConvolution2dLayer(const IConnectableLayer* layer,
+                                                 const Convolution2dDescriptor& convolution2dDescriptor,
+                                                 const ConstTensor& weights,
+                                                 const ConstTensor& biases,
+                                                 const char* name)
+{
+    boost::ignore_unused(biases);
+    VisitConvolution2dLayer(layer, convolution2dDescriptor, weights, name);
+}
+
 void StaticRangeVisitor::VisitActivationLayer(const IConnectableLayer* layer,
                                               const ActivationDescriptor& activationDescriptor,
                                               const char* name)