Rename hheader files to .cuh

This commit is contained in:
2024-02-26 19:53:46 +01:00
parent d3520d5b13
commit 6e99525ad0
10 changed files with 8 additions and 7 deletions

38
include/layers/dense.cuh Normal file
View File

@@ -0,0 +1,38 @@
#ifndef DENSE_LAYER_H
#define DENSE_LAYER_H
#include <vector>
#include <cublas_v2.h>
#include <ilayer.cuh>
namespace Layers {
class Dense : public ILayer {
public:
Dense(int inputSize, int outputSize, cublasHandle_t cublasHandle);
~Dense();
void forward(const float* input, float* output);
void setWeights(const std::vector<std::vector<float>>& weights);
void setBiases(const std::vector<float>& biases);
private:
int inputSize;
int outputSize;
cublasHandle_t cublasHandle;
float* d_weights;
float* d_biases;
std::vector<float> weights;
std::vector<float> biases;
void initializeWeights();
void initializeBiases();
void toCuda();
};
} // namespace Layers
#endif // DENSE_LAYER_H