Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-12-22 14:24:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add avgPool2d implementation

Browse Source
This commit is contained in:
Matúš Námešný
2025-11-21 19:39:30 +01:00
parent c83e1f0c45
commit 5679dc0a50
7 changed files with 102 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
include/backend/cuda.cuh
View File

@@ -49,7 +49,7 @@ class CUDA : public Backend {
const CUDANet::Shape out_shape
) override;
CUDANet::Tensor& CUDA::maxPool2d(
CUDANet::Tensor& maxPool2d(
const CUDANet::Tensor& input,
CUDANet::Tensor& output,
CUDANet::Shape input_shape,
@@ -58,6 +58,16 @@ class CUDA : public Backend {
CUDANet::Shape padding_shape,
CUDANet::Shape output_shape
) override;
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::Backend