Add avgPool2d implementation

2025-12-22 22:34:22 +00:00 · 2025-11-21 19:39:30 +01:00
parent c83e1f0c45
commit 5679dc0a50
7 changed files with 102 additions and 67 deletions
--- a/include/backend.hpp
+++ b/include/backend.hpp
@@ -62,6 +62,16 @@ class Backend {
        CUDANet::Shape padding_shape,
        CUDANet::Shape output_shape
    ) = 0;
+
+    virtual CUDANet::Tensor& avgPool2d(
+        const CUDANet::Tensor& input,
+        CUDANet::Tensor& output,
+        CUDANet::Shape input_shape,
+        CUDANet::Shape pool_shape,
+        CUDANet::Shape stride_shape,
+        CUDANet::Shape padding_shape,
+        CUDANet::Shape output_shape
+    ) = 0;
 };

 }  // namespace CUDANet