servo/AlexNet
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Initial commit

Browse source
This commit is contained in:
Laurent El Shafey 2024-12-10 08:56:11 -08:00
commit 9fdd561586
246 changed files with 58283 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

43
include/quantizer.cuh Normal file
View file

@ -0,0 +1,43 @@
/*
* quantizer.cuh
*
* Created on: 2013-02-15
* Author: spoon
*/
#ifndef QUANTIZER_CUH_
#define QUANTIZER_CUH_
#include <Python.h>
#include <util.cuh>
#include <string>
#include <nvmatrix.cuh>
#include <conv_util.cuh>
class Quantizer {
protected:
NVMatrix* _quantized;
int _numRows, _numCols;
bool _trans;
virtual void _quantize(NVMatrix& src, NVMatrix& tgt);
virtual void _dequantize(NVMatrix& tgt, float scaleTarget, float scaleOutput);
public:
Quantizer();
virtual ~Quantizer();
void quantize(NVMatrix& src, NVMatrix& tgt);
void dequantize(NVMatrix& tgt);
void dequantize(NVMatrix& tgt, float scaleTarget, float scaleOutput);
static Quantizer& make(PyObject* qDict);
};
class HalfQuantizer : public Quantizer {
protected:
void _quantize(NVMatrix& src, NVMatrix& tgt);
void _dequantize(NVMatrix& tgt, float scaleTarget, float scaleOutput);
public:
HalfQuantizer();
};
#endif /* QUANTIZER_CUH_ */