dequantize_linear.h

#ifndef DEQUANTIZE_LINEAR_H
#define DEQUANTIZE_LINEAR_H
 
#include <adf.h>
 
 
template <typename TT, int B, int INP_W, int OUT_W>
class DequantizeLinearScalar {
  
  private:
    float scale;
    TT zero; // same type as output
    
  public:
    DequantizeLinearScalar (
      float scale,
      TT zero
    ): scale(scale), zero(zero) {};
 
        void filter(
            input_window<TT>* in,
            output_window<float>* out
        );
 
        static void registerKernelClass() {
            REGISTER_FUNCTION(DequantizeLinearScalar::filter);
        }
};
 
 
template <typename TT, int B, int INP_W, int OUT_W>
class DequantizeLinear {
  
  private:
    float scale;
    TT zero; // same type as output
 
    // precompute
    int32_t iscale;
    int32_t ishift;
    int bitshift;
    
  public:
    DequantizeLinear (
      float scale,
      TT zero
    );
 
        void filter(
            input_window<TT>* in,
            output_window<float>* out
        );
 
        static void registerKernelClass() {
            REGISTER_FUNCTION(DequantizeLinear::filter);
        }
};
#endif // DEQUANTIZE_LINEAR_H