21 lines
460 B
C++
21 lines
460 B
C++
#ifndef PIANO_DYNASTY_FLOATING_POINT_INCLUDE_COMMON_CUDNNINFERENCER_H_
|
|
#define PIANO_DYNASTY_FLOATING_POINT_INCLUDE_COMMON_CUDNNINFERENCER_H_
|
|
|
|
#include "PianoInferencer.h"
|
|
|
|
namespace dynasty {
|
|
namespace inferencer {
|
|
namespace cudnn {
|
|
|
|
template <typename T>
|
|
class Inferencer : public PianoInferencer<T> {
|
|
public:
|
|
static typename PianoInferencer<T>::Builder* GetBuilder();
|
|
};
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif //PIANO_DYNASTY_FLOATING_POINT_INCLUDE_COMMON_CUDNNINFERENCER_H_
|