24 : m_InputNetwork(inputNetwork),
50 unsigned int m_RefineCount;
58 std::pair<float, float> FindMinMax(
ITensorHandle* tensorHandle);
NetworkQuantizer(INetwork *inputNetwork, const QuantizerOptions &options)
std::unique_ptr< IRuntime, void(*)(IRuntime *runtime)> IRuntimePtr
Main network class which provides the interface for building up a neural network. ...
std::vector< std::pair< LayerBindingId, class ConstTensor > > InputTensors
Copyright (c) 2020 ARM Limited.
int LayerBindingId
Type of identifiers for bindable layers (inputs, outputs).
MinMaxRange GetRange(LayerGuid guid, unsigned int idx) const
Retrieve the Range for a particular output slot on a particular layer.
static void Destroy(INetworkQuantizer *quantizer)
Destroy Quantizer object.
std::pair< float, float > GetMinMaxRange(LayerGuid guid, unsigned int idx)
void Refine(const InputTensors &inputTensors) override
Refine input network with a set of refinement data for specified LayerBindingId.
void OverrideInputRange(LayerBindingId layerId, float min, float max) override
Overrides the default quantization values for the input layer with the given id.
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr
Quantizer class Quantizes a float32 InputNetwork.
armnn::Runtime::CreationOptions::ExternalProfilingOptions options
INetworkPtr ExportNetwork() override
Extract final quantized network.