Set up cmake to compile library

This commit is contained in:
2024-02-17 23:07:09 +01:00
parent ac18768297
commit f541e2f7f8
7 changed files with 40 additions and 93 deletions

View File

@@ -3,6 +3,7 @@
#define I_LAYER_H
#include <cublas_v2.h>
#include <vector>
namespace Layers {
@@ -11,6 +12,8 @@ namespace Layers {
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