aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/NetworkQuantizer.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/NetworkQuantizer.hpp')
-rw-r--r--src/armnn/NetworkQuantizer.hpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/src/armnn/NetworkQuantizer.hpp b/src/armnn/NetworkQuantizer.hpp
index 5b87851edf..5e93f70290 100644
--- a/src/armnn/NetworkQuantizer.hpp
+++ b/src/armnn/NetworkQuantizer.hpp
@@ -17,7 +17,8 @@ namespace armnn
class NetworkQuantizer : public INetworkQuantizer
{
public:
- NetworkQuantizer(INetwork* inputNetwork) : m_InputNetwork(inputNetwork) {}
+ NetworkQuantizer(INetwork* inputNetwork, const QuantizerOptions& options)
+ : m_InputNetwork(inputNetwork), m_Options(options) {}
void OverrideInputRange(LayerBindingId layerId, float min, float max) override;
INetworkPtr ExportNetwork() override;
@@ -28,6 +29,9 @@ private:
/// Mapping from Guid to an array of ranges for outputs
RangeTracker m_Ranges;
+
+ /// Options for the NetworkQuantizer
+ QuantizerOptions m_Options;
};
} //namespace armnn