1 files changed, 47 insertions, 0 deletions
diff --git a/src/armnn/OverrideInputRangeVisitor.cpp b/src/armnn/OverrideInputRangeVisitor.cpp
new file mode 100644
index 0000000000..4c70d3f4a6
--- /dev/null
+++ b/src/armnn/OverrideInputRangeVisitor.cpp
@@ -0,0 +1,47 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "OverrideInputRangeVisitor.hpp"
+#include "NetworkQuantizerUtils.hpp"
+#include "Layer.hpp"
+
+#include <boost/assert.hpp>
+
+namespace armnn
+{
+
+OverrideInputRangeVisitor::OverrideInputRangeVisitor(std::unordered_map<LayerGuid, MinMaxRanges>& guidToRangesMap,
+                                                     LayerBindingId layerId,
+                                                     const MinMaxRange& minMaxRange)
+    : m_GuidToRangesMap(guidToRangesMap)
+    , m_LayerId(layerId)
+    , m_MinMaxRange(minMaxRange)
+{}
+
+void OverrideInputRangeVisitor::VisitInputLayer(const IConnectableLayer *layer, LayerBindingId id, const char *name)
+{
+    if (m_LayerId != id)
+    {
+        // Not the layer we are looking for
+        return;
+    }
+
+    SetRange(layer);
+}
+
+void OverrideInputRangeVisitor::SetRange(const IConnectableLayer* layer)
+{
+    BOOST_ASSERT(layer);
+
+    auto& ranges = m_GuidToRangesMap[layer->GetGuid()];
+
+    if (ranges.size() < layer->GetNumOutputSlots())
+    {
+        ranges.resize(layer->GetNumOutputSlots());
+    }
+    ranges[0] = m_MinMaxRange;
+}
+
+} // namespace armnn