Initial cpu conv implementation

This commit is contained in:
2024-03-07 21:24:59 +01:00
parent 07f231a30b
commit fc2c1616b4
2 changed files with 58 additions and 14 deletions

View File

@@ -49,6 +49,10 @@ class Conv2d {
void initializeKernels();
void toCuda();
void setKernels(const std::vector<float>& kernels_input);
void host_conv(const float* input, float* output);
};
} // namespace Layers