Skip to content
Snippets Groups Projects

Forked from add_matmul (merged automatically)

Merged Benjamin Halimi requested to merge quantizer_only into dev
10 files
+ 448
292
Compare changes
  • Side-by-side
  • Inline
Files
10
@@ -9,30 +9,30 @@
*
********************************************************************************/
#ifndef AIDGE_QUANTIZATION_QUANTIZATION_QAT_LSQ_H_
#define AIDGE_QUANTIZATION_QUANTIZATION_QAT_LSQ_H_
#include <cstddef> // std::size_t
#include <memory>
#include "aidge/data/Tensor.hpp"
#include "aidge/graph/GraphView.hpp"
namespace Aidge {
namespace QuantLSQ {
/**
* @brief Given a GraphView with parameters properly initialized, insert
* the LSQ quantizer nodes, and setup the adjustment their step-sizes.
* @param graphView The GraphView containing the network to quantize.
* @param nbBits Number of quantization bits.
*/
void setupQuantizers(std::shared_ptr<GraphView> graphView, size_t nbBits);
} // namespace QuantLSQ
} // namespace Aidge
#endif /* AIDGE_QUANTIZATION_QUANTIZATION_QAT_LSQ_H_ */
#ifndef AIDGE_QUANTIZATION_QUANTIZATION_QAT_LSQ_H_
#define AIDGE_QUANTIZATION_QUANTIZATION_QAT_LSQ_H_
#include <cstddef> // std::size_t
#include <memory>
#include "aidge/data/Tensor.hpp"
#include "aidge/graph/GraphView.hpp"
namespace Aidge {
namespace QuantLSQ {
/**
* @brief Given a GraphView with parameters properly initialized, insert
* the LSQ quantizer nodes, and setup the adjustment their step-sizes.
* @param graphView The GraphView containing the network to quantize.
* @param nbBits Number of quantization bits.
*/
void setupQuantizers(std::shared_ptr<GraphView> graphView, size_t nbBits);
} // namespace QuantLSQ
} // namespace Aidge
#endif /* AIDGE_QUANTIZATION_QUANTIZATION_QAT_LSQ_H_ */
\ No newline at end of file
Loading