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();
55 network->Accept(inputLayerVisitor);
60 std::vector<std::vector<float>> inputData(quantizationInput.GetNumberOfInputs());
61 std::vector<armnn::LayerBindingId> layerBindingIds = quantizationInput.GetLayerBindingIds();
62 unsigned int count = 0;
66 inputData[count] = quantizationInput.GetDataForEntry(layerBindingId);
68 inputTensors.push_back(std::make_pair(layerBindingId, inputTensor));
71 quantizer->Refine(inputTensors);
78 serializer->Serialize(*quantizedNetwork);
82 std::ofstream outputFileStream;
83 outputFileStream.open(output);
84 serializer->SaveSerializedToStream(outputFileStream);
85 outputFileStream.flush();
86 outputFileStream.close();
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
bool ProcessCommandLine(int argc, char *argv[])
std::string GetOutputFileName()
std::unique_ptr< ISerializer, void(*)(ISerializer *serializer)> ISerializerPtr
bool HasPreservedDataType()
std::unique_ptr< class INetworkQuantizer, void(*)(INetworkQuantizer *quantizer)> INetworkQuantizerPtr
std::vector< std::pair< LayerBindingId, class ConstTensor > > InputTensors
std::string GetQuantizationScheme()
static ISerializerPtr Create()
std::unique_ptr< IDeserializer, void(*)(IDeserializer *parser)> IDeserializerPtr
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr
std::string GetOutputDirectoryName()
static IDeserializerPtr Create()
static INetworkQuantizerPtr Create(INetwork *inputNetwork, const QuantizerOptions &options=QuantizerOptions())
Create Quantizer object wrapped in unique_ptr.
bool HasQuantizationData()
DataType m_ActivationFormat
QuantizationDataSet GetQuantizationDataSet()
int main(int argc, char *argv[])
int LayerBindingId
Type of identifiers for bindable layers (inputs, outputs).
std::string GetInputFileName()