From a40521a70e73d20a060fa2df0e83b02c4f1c6139 Mon Sep 17 00:00:00 2001
From: ruoyan01 <ruomei.yan@arm.com>
Date: Mon, 11 Feb 2019 13:22:32 +0000
Subject: IVGCVSW-2625 Support static quantization of softmax

Change-Id: I216344ee10bbffadb648d4aef1d9a0d9dbb4a341
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Signed-off-by: ruoyan01 <ruomei.yan@arm.com>
---
 src/armnn/StaticRangeVisitor.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src/armnn/StaticRangeVisitor.cpp')

diff --git a/src/armnn/StaticRangeVisitor.cpp b/src/armnn/StaticRangeVisitor.cpp
index 6ae0d3314b..05117cd5cd 100644
--- a/src/armnn/StaticRangeVisitor.cpp
+++ b/src/armnn/StaticRangeVisitor.cpp
@@ -113,4 +113,12 @@ void StaticRangeVisitor::VisitFullyConnectedLayer(const IConnectableLayer *layer
     SetRange(layer, 0, -15.0f, 15.0f);
 }
 
+void StaticRangeVisitor::VisitSoftmaxLayer(const IConnectableLayer* layer,
+                                           const SoftmaxDescriptor& softmaxDescriptor,
+                                           const char* name)
+{
+    boost::ignore_unused(softmaxDescriptor);
+    SetRange(layer, 0, 0.f, 1.f);
+}
+
 } //namespace armnn
-- 
cgit v1.2.1