Change forward function to return output pointer

2025-11-06 01:34:22 +00:00 · 2024-03-12 20:50:49 +01:00
parent 2518138ef8
commit 9d91896f13
8 changed files with 69 additions and 113 deletions
--- a/include/layers/dense.cuh
+++ b/include/layers/dense.cuh
@@ -18,7 +18,7 @@ class Dense : public ILayer {
    );
    ~Dense();

-    void forward(const float* d_input, float* d_output);
+    float* forward(const float* d_input);
    void setWeights(const float* weights);
    void setBiases(const float* biases);

@@ -26,6 +26,8 @@ class Dense : public ILayer {
    int inputSize;
    int outputSize;

+    float* d_output;
+
    float* d_weights;
    float* d_biases;