Add bias to conv2d

This commit is contained in:
2024-03-09 23:03:23 +01:00
parent 4f3c4f1afb
commit d177a67cd6
2 changed files with 28 additions and 4 deletions

View File

@@ -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();
};