mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 09:44:28 +00:00
Initial cpu conv implementation
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user