mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-05 17:34:21 +00:00
Add bias to conv2d
This commit is contained in:
@@ -42,15 +42,18 @@ class Conv2d {
|
||||
|
||||
// Kernels
|
||||
std::vector<float> kernels;
|
||||
std::vector<float> biases;
|
||||
|
||||
// Cuda
|
||||
float* d_kernels;
|
||||
float* d_biases;
|
||||
float* d_padded;
|
||||
|
||||
// Kernels
|
||||
Activation activation;
|
||||
|
||||
void initializeKernels();
|
||||
void initializeBiases();
|
||||
void toCuda();
|
||||
};
|
||||
|
||||
|
||||
Reference in New Issue
Block a user