graph_argmax.h

#ifndef __ARGMAX_GRAPH_H__
#define __ARGMAX_GRAPH_H__
 
#include <adf.h>
#include "argmax.h"
 
 
template <template<int, int> class ARGMAX, int CHUNK_CNT, int CHUNK_SIZE>
class ArgmaxGraph : public adf::graph {
 
  private:
    adf::kernel k[1];
    std::string id;
 
  public:
    adf::port<input> pin[1];
    adf::port<output> pout[1];
 
    ArgmaxGraph() { 
      k[0] = adf::kernel::create_object<ARGMAX<CHUNK_CNT, CHUNK_SIZE>>();
      adf::source(k[0]) = "argmax.cc";
      adf::headers(k[0]) = {"argmax.h"};
      adf::runtime<ratio>(k[0]) = 0.6;
      
      adf::connect<adf::window<CHUNK_CNT*CHUNK_SIZE*4>> (pin[0], k[0].in[0]);
      adf::connect<adf::window<CHUNK_CNT*4>> (k[0].out[0], pout[0]);
    }
 
};
#endif // __ARGMAX_GRAPH_H__