transpose.h

#ifndef TRANSPOSE_H_
#define TRANSPOSE_H_
 
#include <adf.h>
 
 
template <typename TT, int B, int H, int W, int C, int PAD_W>
class TransposeScalarBHWC2BCHW {
  public:
    void filter(
      input_window<TT>* in,
      output_window<TT>* out
    );
 
    static void registerKernelClass() {
      static_assert(W == PAD_W);
      REGISTER_FUNCTION(TransposeScalarBHWC2BCHW::filter);
    }
};
 
 
template <typename TT, int B, int H, int W, int C, int PAD_W>
class TransposeScalarBCHW2BHWC {
  public:
    void filter(
      input_window<TT>* in,
      output_window<TT>* out
    );
 
    static void registerKernelClass() {
      REGISTER_FUNCTION(TransposeScalarBCHW2BHWC::filter);
    }
};
 
 
template <typename TT, int B, int H, int W, int C, int PAD_W>
class TransposeScalarBHWC2BCHWStream {
  public:
    void filter(
      input_window<TT>* in,
      output_stream<TT>* restrict out
    );
 
    static void registerKernelClass() {
      static_assert(sizeof(TT) == 4);
      REGISTER_FUNCTION(TransposeScalarBHWC2BCHWStream::filter);
    }
};
 
template <typename TT, int B, int H, int W, int C, int PAD_W>
class TransposeScalarPktStreamBHWC2BCHW {
  public:
    void filter(
      input_pktstream* in,
      output_window<TT>* out
    );
 
    static void registerKernelClass() {
      REGISTER_FUNCTION(TransposeScalarPktStreamBHWC2BCHW::filter);
    }
};
#endif // TRANSPOSE_H_