Add Kernels namespace

This commit is contained in:
2024-03-11 21:04:23 +01:00
parent e0178e2d5c
commit d2ab78fbc7
18 changed files with 188 additions and 186 deletions

View File

@@ -5,7 +5,7 @@
#include <vector>
#include "activations.cuh"
#include "padding.cuh"
#include "convolution.cuh"
#include "ilayer.cuh"
namespace Layers {
@@ -13,13 +13,13 @@ namespace Layers {
class Conv2d : public ILayer {
public:
Conv2d(
int inputSize,
int inputChannels,
int kernelSize,
int stride,
Padding padding,
int numFilters,
Activation activation
int inputSize,
int inputChannels,
int kernelSize,
int stride,
Layers::Padding padding,
int numFilters,
Layers::Activation activation
);
~Conv2d();
@@ -52,7 +52,7 @@ class Conv2d : public ILayer {
float* d_padded;
// Kernels
Activation activation;
Layers::Activation activation;
void initializeWeights();
void initializeBiases();