Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 09:44:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Format source code using clang-format

Browse Source
This commit is contained in:
Matúš Námešný
2024-02-27 18:51:22 +01:00
parent fb454de053
commit 48ba09b28d
12 changed files with 229 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
include/layers/conv.cuh
View File

@@ -5,22 +5,27 @@
namespace Layers {
class Conv {
public:
Conv(int inputSize, int outputSize, int kernelSize, cublasHandle_t cublasHandle);
~Conv();
class Conv {
public:
Conv(
int inputSize,
int outputSize,
int kernelSize,
cublasHandle_t cublasHandle
);
~Conv();
void forward(const float* input, float* output);
void forward(const float* input, float* output);
private:
int inputSize;
int outputSize;
int kernelSize;
cublasHandle_t cublasHandle;
float* d_weights;
float* d_biases;
};
private:
int inputSize;
int outputSize;
int kernelSize;
cublasHandle_t cublasHandle;
float* d_weights;
float* d_biases;
};
} // namespace Layers
} // namespace Layers
#endif // CONV_LAYER_H
#endif // CONV_LAYER_H