mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
18 lines
261 B
C++
18 lines
261 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;
|
|
};
|
|
|
|
} // namespace Layers
|
|
|
|
#endif // I_LAYERH
|