Migrate add layer to tensors

2025-12-22 14:24:22 +00:00 · 2025-11-22 00:12:20 +01:00
parent aeb1739c46
commit 4c8b2ef537
7 changed files with 56 additions and 101 deletions
--- a/include/backend.hpp
+++ b/include/backend.hpp
@@ -90,6 +90,12 @@ class Backend {
        CUDANet::Tensor& input_b,
        CUDANet::Tensor& output
    ) = 0;
+
+    virtual CUDANet::Tensor& add(
+        CUDANet::Tensor& input_a,
+        CUDANet::Tensor& input_b,
+        CUDANet::Tensor& output
+    ) = 0;
 };

 }  // namespace CUDANet