ArmNN
NotReleased
|
#include <NetworkQuantizer.hpp>
Public Member Functions | |
NetworkQuantizer (INetwork *inputNetwork, const QuantizerOptions &options) | |
void | OverrideInputRange (LayerBindingId layerId, float min, float max) override |
Overrides the default quantization values for the input layer with the given id. More... | |
void | Refine (const InputTensors &inputTensors) override |
Refine input network with a set of refinement data for specified LayerBindingId. More... | |
std::pair< float, float > | GetMinMaxRange (LayerGuid guid, unsigned int idx) |
INetworkPtr | ExportNetwork () override |
Extract final quantized network. More... | |
Additional Inherited Members | |
Static Public Member Functions inherited from INetworkQuantizer | |
static INetworkQuantizer * | CreateRaw (INetwork *inputNetwork, const QuantizerOptions &options=QuantizerOptions()) |
Create Quantizer object and return raw pointer. More... | |
static INetworkQuantizerPtr | Create (INetwork *inputNetwork, const QuantizerOptions &options=QuantizerOptions()) |
Create Quantizer object wrapped in unique_ptr. More... | |
static void | Destroy (INetworkQuantizer *quantizer) |
Destroy Quantizer object. More... | |
Protected Member Functions inherited from INetworkQuantizer | |
virtual | ~INetworkQuantizer () |
Definition at line 20 of file NetworkQuantizer.hpp.
|
inline |
Definition at line 23 of file NetworkQuantizer.hpp.
References NetworkQuantizer::OverrideInputRange(), and NetworkQuantizer::Refine().
|
overridevirtual |
Extract final quantized network.
Implements INetworkQuantizer.
Definition at line 136 of file NetworkQuantizer.cpp.
References armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, Graph::TopologicalSort(), and armnn::VisitLayers().
Referenced by NetworkQuantizer::GetMinMaxRange().
|
inline |
Definition at line 34 of file NetworkQuantizer.hpp.
References NetworkQuantizer::ExportNetwork(), and RangeTracker::GetRange().
|
overridevirtual |
Overrides the default quantization values for the input layer with the given id.
Implements INetworkQuantizer.
Definition at line 50 of file NetworkQuantizer.cpp.
References Graph::GetInputLayers(), and armnn::VisitLayers().
Referenced by NetworkQuantizer::NetworkQuantizer().
|
overridevirtual |
Refine input network with a set of refinement data for specified LayerBindingId.
Implements INetworkQuantizer.
Definition at line 60 of file NetworkQuantizer.cpp.
References IRuntime::Create(), armnnUtils::FindMinMax(), armnn::Optimize(), options, Graph::TopologicalSort(), and armnn::VisitLayers().
Referenced by NetworkQuantizer::NetworkQuantizer().