graph_dequantize_linear.h

#ifndef __DEQUANTIZE_LINEAR_GRAPH_H__
#define __DEQUANTIZE_LINEAR_GRAPH_H__
 
#include <assert.h>
#include <adf.h>
#include "dequantize_linear.h"
 
 
template <template<typename, int, int, int> class DEQUANTIZE_LINEAR, 
  typename TT, int B, int INP_W, int OUT_W>
class DequantizeLinearGraph : public adf::graph {
 
  private:
    adf::kernel k[1];
    std::string id;
 
  public:
    adf::port<input> pin[1];
    adf::port<output> pout[1];
 
    DequantizeLinearGraph(
      float scale,
      TT zero,
      int repeat_cnt = 1
    ) { 
      static_assert(INP_W >= OUT_W);
      k[0] = adf::kernel::create_object<DEQUANTIZE_LINEAR<TT, B, INP_W, OUT_W>>(scale, zero);
      adf::source(k[0]) = "dequantize_linear.cc";
      adf::headers(k[0]) = {"dequantize_linear.h"};
      adf::runtime<ratio>(k[0]) = 0.6;
      adf::repetition_count(k[0]) = repeat_cnt;
      
      adf::connect<adf::window<B*INP_W>> (pin[0], k[0].in[0]);
      adf::connect<adf::window<B*OUT_W*4>> (k[0].out[0], pout[0]);
    }
 
};
#endif // __DEQUANTIZE_LINEAR_GRAPH_H__