qlinearadd.h

#ifndef QLINEARADD_H_
#define QLINEARADD_H_
 
#include <assert.h>
#include <type_traits>
#include <adf.h>
 
 
template <typename TT, int W, int IS_RELU>
class QLinearAddInt8 {
    private:
    float a_scale;
    float b_scale;
    float c_scale;
    TT a_zero;
    TT b_zero;
    TT c_zero;
 
        int bitshift;
        int32_t ascale;
        int32_t bscale;
        int32_t shiftv;
    
    public:
        QLinearAddInt8 (
      float a_scale,
            float b_scale,
            float c_scale,
            TT a_zero,
            TT b_zero,
            TT c_zero
    );
 
        void filter(
            input_stream<TT>* restrict inA,
            input_stream<TT>* restrict inB,
            output_stream<TT>* restrict out
        );
 
        static void registerKernelClass() {
            static_assert(W%16==0);
            static_assert((std::is_same<TT, int8_t>::value) || (std::is_same<TT, uint8_t>::value));
            REGISTER_FUNCTION(QLinearAddInt8::filter);
        }
};
#endif // QLINEARADD_H_