Fix some dense layer issues

2025-12-22 22:34:22 +00:00 · 2025-11-18 22:17:08 +01:00
parent 7f203b8947
commit 4c26efe826
8 changed files with 110 additions and 44 deletions
--- a/include/backend.hpp
+++ b/include/backend.hpp
@@ -32,12 +32,12 @@ class Backend {
    ) = 0;

    virtual CUDANet::Tensor& dense(
-        CUDANet::Tensor& weights,
-        CUDANet::Tensor& biases,
-        CUDANet::Tensor& input,
+        const CUDANet::Tensor& weights,
+        const CUDANet::Tensor& biases,
+        const CUDANet::Tensor& input,
        CUDANet::Tensor& output,
-        size_t           input_size,
-        size_t           output_size
+        const size_t           input_size,
+        const size_t           output_size
    ) = 0;
 };