17 int main(
int argc,
char* argv[])
26 std::vector<std::uint8_t> binaryContent;
27 while (inputFileStream)
30 inputFileStream.get(c);
33 binaryContent.push_back(static_cast<std::uint8_t>(c));
36 inputFileStream.close();
65 network->Accept(inputLayerVisitor);
70 std::vector<std::vector<float>> inputData(quantizationInput.GetNumberOfInputs());
71 std::vector<armnn::LayerBindingId> layerBindingIds = quantizationInput.GetLayerBindingIds();
72 unsigned int count = 0;
76 inputData[count] = quantizationInput.GetDataForEntry(layerBindingId);
78 inputTensors.push_back(std::make_pair(layerBindingId, inputTensor));
81 quantizer->Refine(inputTensors);
88 serializer->Serialize(*quantizedNetwork);
92 std::ofstream outputFileStream;
93 outputFileStream.open(output);
94 serializer->SaveSerializedToStream(outputFileStream);
95 outputFileStream.flush();
96 outputFileStream.close();
bool HasQuantizationData()
std::string GetOutputFileName()
DataType m_ActivationFormat
QuantizationDataSet is a structure which is created after parsing a quantization CSV file...
bool ProcessCommandLine(int argc, char *argv[])
std::unique_ptr< class INetworkQuantizer, void(*)(INetworkQuantizer *quantizer)> INetworkQuantizerPtr
static IDeserializerPtr Create()
std::vector< std::pair< LayerBindingId, class ConstTensor > > InputTensors
int LayerBindingId
Type of identifiers for bindable layers (inputs, outputs).
bool HasPreservedDataType()
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
std::string GetQuantizationScheme()
std::unique_ptr< IDeserializer, void(*)(IDeserializer *parser)> IDeserializerPtr
int main(int argc, char *argv[])
std::string GetInputFileName()
std::unique_ptr< ISerializer, void(*)(ISerializer *serializer)> ISerializerPtr
std::string GetOutputDirectoryName()
QuantizationDataSet GetQuantizationDataSet()
static ISerializerPtr Create()
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr
static INetworkQuantizerPtr Create(INetwork *inputNetwork, const QuantizerOptions &options=QuantizerOptions())
Create Quantizer object wrapped in unique_ptr.