mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
32 lines
582 B
Plaintext
32 lines
582 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
|