mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-05 17:34:21 +00:00
22 lines
442 B
Plaintext
22 lines
442 B
Plaintext
|
|
#ifndef I_LAYER_H
|
|
#define I_LAYER_H
|
|
|
|
#include <cublas_v2.h>
|
|
|
|
#include <vector>
|
|
|
|
namespace Layers {
|
|
|
|
class ILayer {
|
|
public:
|
|
virtual ~ILayer() {}
|
|
|
|
virtual void forward(const float* input, float* output) = 0;
|
|
virtual void setWeights(const std::vector<std::vector<float>>& weights) = 0;
|
|
virtual void setBiases(const std::vector<float>& biases) = 0;
|
|
};
|
|
|
|
} // namespace Layers
|
|
|
|
#endif // I_LAYERH |