Migrate MaxPool2d layer to Tensors

2025-12-22 14:24:22 +00:00 · 2025-11-19 21:44:19 +01:00
parent 7896ff0e24
commit e4d05931d4
17 changed files with 215 additions and 454 deletions
--- a/include/backend.hpp
+++ b/include/backend.hpp
@@ -52,6 +52,16 @@ class Backend {
        const CUDANet::Shape stride_shape,
        const CUDANet::Shape out_shape
    ) = 0;
+
+    virtual CUDANet::Tensor& maxPool2d(
+        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