graph_softmax.h

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