From 27d830720ed706f187f2a40e2c5055e424aa8b91 Mon Sep 17 00:00:00 2001
From: Derek Lamberti <derek.lamberti@arm.com>
Date: Tue, 5 Feb 2019 16:00:08 +0000
Subject: IVGCVSW-2606 Produce quantized InputNetwork from simple FP32
 InputNetwork

Change-Id: I2140a7af5961ddf8267fbb127202de3900ea79e3
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
---
 include/armnn/INetworkQuantizer.hpp | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 include/armnn/INetworkQuantizer.hpp

(limited to 'include')

diff --git a/include/armnn/INetworkQuantizer.hpp b/include/armnn/INetworkQuantizer.hpp
new file mode 100644
index 0000000000..f3cc13ca35
--- /dev/null
+++ b/include/armnn/INetworkQuantizer.hpp
@@ -0,0 +1,33 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <armnn/INetwork.hpp>
+
+#include <memory>
+
+
+namespace armnn
+{
+
+using INetworkQuantizerPtr = std::unique_ptr<class INetworkQuantizer, void(*)(INetworkQuantizer* quantizer)>;
+
+/// Quantizer class Quantizes a float32 InputNetwork
+class INetworkQuantizer
+{
+public:
+    static INetworkQuantizer* CreateRaw(INetwork* inputNetwork); ///< Create Quantizer object and return raw pointer
+    static INetworkQuantizerPtr Create(INetwork* inputNetwork);  ///< Create Quantizer object wrapped in unique_ptr
+    static void Destroy(INetworkQuantizer* quantizer);           ///< Destroy Quantizer object
+
+    /// Extract final quantized network
+    virtual INetworkPtr ExportNetwork() = 0;
+
+protected:
+    virtual ~INetworkQuantizer() {};
+};
+
+} //namespace armnn
-- 
cgit v1.2.1