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

28
include/layers/conv.cuh 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