Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-07 18:24:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change forward function to return output pointer

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-12 20:50:49 +01:00
parent 2518138ef8
commit 9d91896f13
8 changed files with 69 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
include/layers/conv2d.cuh
View File

@@ -26,10 +26,10 @@ class Conv2d : public ILayer {
// Outputs
int outputSize;
void forward(const float* d_input, float* d_output);
void setWeights(const float* weights_input);
void setBiases(const float* biases_input);
void host_conv(const float* input, float* output);
float* forward(const float* d_input);
void setWeights(const float* weights_input);
void setBiases(const float* biases_input);
void host_conv(const float* input, float* output);
private:
// Inputs
@@ -47,6 +47,7 @@ class Conv2d : public ILayer {
std::vector<float> biases;
// Cuda
float* d_output;
float* d_weights;
float* d_biases;
float* d_padded;