diff options
Diffstat (limited to 'src/armnn/NetworkQuantizer.hpp')
-rw-r--r-- | src/armnn/NetworkQuantizer.hpp | 6 |
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 |