diff options
Diffstat (limited to 'include/armnn/INetworkQuantizer.hpp')
| -rw-r--r-- | include/armnn/INetworkQuantizer.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/include/armnn/INetworkQuantizer.hpp b/include/armnn/INetworkQuantizer.hpp index 54c1c889d3..89548d1057 100644 --- a/include/armnn/INetworkQuantizer.hpp +++ b/include/armnn/INetworkQuantizer.hpp @@ -7,6 +7,7 @@ #include <armnn/INetwork.hpp> #include <armnn/Types.hpp> +#include <armnn/Tensor.hpp> namespace armnn { @@ -37,6 +38,9 @@ public: /// Overrides the default quantization values for the input layer with the given id virtual void OverrideInputRange(LayerBindingId layerId, float min, float max) = 0; + /// Refine input network with a set of refinement data for specified LayerBindingId + virtual void Refine(const InputTensors& inputTensors) = 0; + /// Extract final quantized network virtual INetworkPtr ExportNetwork() = 0; |