Implement to_cuda function

This commit is contained in:
2024-02-09 19:18:30 +01:00
parent 6645cd31ba
commit 3e6b7dc7e6
4 changed files with 37 additions and 6 deletions

View File

@@ -5,15 +5,17 @@
#include <vector>
#include <cublas_v2.h>
#include <ilayer.h>
namespace Layers {
class Dense {
class Dense : public ILayer {
public:
Dense(int inputSize, int outputSize, cublasHandle_t cublasHandle);
~Dense();
void forward(const float* input, float* output);
void to_cuda();
private:
int inputSize;