mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-07 18:24:26 +00:00
19 lines
297 B
C++
19 lines
297 B
C++
|
|
#ifndef I_LAYER_H
|
|
#define I_LAYER_H
|
|
|
|
#include <cublas_v2.h>
|
|
|
|
namespace Layers {
|
|
|
|
class ILayer {
|
|
public:
|
|
virtual ~ILayer() {}
|
|
|
|
virtual void forward(const float* input, float* output) = 0;
|
|
virtual void to_cuda() = 0;
|
|
};
|
|
|
|
} // namespace Layers
|
|
|
|
#endif // I_LAYERH
|