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/cuda.cuh
+++ b/include/backend/cuda.cuh
@@ -48,6 +48,16 @@ class CUDA : public Backend {
        const CUDANet::Shape stride_shape,
        const CUDANet::Shape out_shape
    ) override;
+
+    CUDANet::Tensor& CUDA::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
+    ) override;
 };

 }  // namespace CUDANet::Backend