Start dense layer implementation

This commit is contained in:
2024-02-08 19:19:51 +01:00
parent b16ec69469
commit 950021389c
6 changed files with 121 additions and 12 deletions

28
include/layers/conv.h Normal file
View File

@@ -0,0 +1,28 @@
// fully_connected_layer.h
#ifndef CONV_LAYER_H
#define CONV_LAYER_H
#include <cublas_v2.h>
namespace Layers {
class Conv {
public:
Conv(int inputSize, int outputSize, int kernelSize, cublasHandle_t cublasHandle);
~Conv();
void forward(const float* input, float* output);
private:
int inputSize;
int outputSize;
int kernelSize;
cublasHandle_t cublasHandle;
float* d_weights;
float* d_biases;
};
} // namespace Layers
#endif // CONV_LAYER_H