Skip to content
Snippets Groups Projects

Minor Changes

Merged Benjamin Halimi requested to merge bhalimi/aidge_quantization:DevPTQ into main
Files
4
@@ -64,7 +64,7 @@ namespace Aidge {
* @param graphView The GraphView to be quantized.
* @param nbBits The desired number of bits of the quantization.
*/
void quantizeNormalizedNetwork(std::shared_ptr<GraphView> graphView, int nbBits);
void quantizeNormalizedNetwork(std::shared_ptr<GraphView> graphView, std::uint8_t nbBits);
/**
* @brief Main quantization routine. Performs every step of the quantization pipeline.
@@ -72,7 +72,7 @@ namespace Aidge {
* @param nbBits The desired number of bits of the quantization.
* @param inputDataSet The input dataset on which the value ranges are computed.
*/
void quantizeNetwork(std::shared_ptr<GraphView> graphView, int nbBits, std::vector<std::shared_ptr<Tensor>> inputDataSet);
void quantizeNetwork(std::shared_ptr<GraphView> graphView, std::uint8_t nbBits, std::vector<std::shared_ptr<Tensor>> inputDataSet);
/**
* @brief Compute the weight ranges of every affine node. Provided for debuging purposes.
Loading