Start conv test implementation

This commit is contained in:
2024-03-07 22:03:05 +01:00
parent 7e75943a6b
commit 69ccba2dad
3 changed files with 48 additions and 12 deletions

View File

@@ -11,16 +11,19 @@ namespace Layers {
class Conv2d {
public:
Conv2d(
int inputSize,
int inputChannels,
int kernelSize,
int stride,
std::string padding,
int numFilters,
Activation activation
int inputSize,
int inputChannels,
int kernelSize,
int stride,
std::string padding,
int numFilters,
Activation activation
);
~Conv2d();
// Outputs
int outputSize;
void forward(const float* d_input, float* d_output);
private:
@@ -34,15 +37,12 @@ class Conv2d {
int paddingSize;
int numFilters;
// Outputs
int outputSize;
// Kernels
std::vector<float> kernels;
// Cuda
float* d_kernels;
float* d_padded;
float* d_kernels;
float* d_padded;
// Kernels
Activation activation;