IVGCVSW-2614 Support static quantization of Permute

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

1 files changed, 14 insertions, 0 deletions

diff --git a/src/armnn/StaticRangeVisitor.cpp b/src/armnn/StaticRangeVisitor.cpp
index 2365e1b38c..ecdc1ad957 100644
--- a/src/armnn/StaticRangeVisitor.cpp
+++ b/src/armnn/StaticRangeVisitor.cpp
@@ -21,6 +21,12 @@ void StaticRangeVisitor::SetRange(const IConnectableLayer* layer, unsigned int o
     m_RangeTracker.SetRange(layer, outputIdx, min, max);
 }
 
+void StaticRangeVisitor::ForwardParentParameters(const IConnectableLayer* layer)
+{
+    const auto parentRange = m_RangeTracker.GetRange(layer->GetInputSlot(0).GetConnection()->GetOwningLayerGuid(), 0);
+    SetRange(layer, 0, parentRange.first, parentRange.second);
+}
+
 void StaticRangeVisitor::VisitAdditionLayer(const IConnectableLayer* layer, const char* name)
 {
     SetRange(layer, 0, -20.f, 20.f);
@@ -110,6 +116,14 @@ void StaticRangeVisitor::VisitFullyConnectedLayer(const IConnectableLayer *layer
     SetRange(layer, 0, -15.0f, 15.0f);
 }
 
+void StaticRangeVisitor::VisitPermuteLayer(const IConnectableLayer* layer,
+                                           const PermuteDescriptor& permuteDescriptor,
+                                           const char* name)
+{
+    boost::ignore_unused(permuteDescriptor);
+    ForwardParentParameters(layer);
+}
+
 void StaticRangeVisitor::VisitSoftmaxLayer(const IConnectableLayer* layer,
                                            const SoftmaxDescriptor& softmaxDescriptor,
                                            const char* name)