mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-05 17:34:21 +00:00
27 lines
508 B
Plaintext
27 lines
508 B
Plaintext
#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
|