Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 09:44:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename dim2d to shape2d

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-27 21:14:51 +02:00
parent 07d505a0e5
commit df47a31f36
22 changed files with 120 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
include/kernels/pooling.cuh
View File

@@ -9,23 +9,23 @@ namespace CUDANet::Kernels {
__global__ void max_pooling(
const float* __restrict__ d_input,
float* __restrict__ d_output,
const dim2d inputSize,
const dim2d outputSize,
const shape2d inputSize,
const shape2d outputSize,
const int nChannels,
const dim2d poolingSize,
const dim2d stride,
const dim2d padding
const shape2d poolingSize,
const shape2d stride,
const shape2d padding
);
__global__ void avg_pooling(
const float* __restrict__ d_input,
float* __restrict__ d_output,
const dim2d inputSize,
const dim2d outputSize,
const shape2d inputSize,
const shape2d outputSize,
const int nChannels,
const dim2d poolingSize,
const dim2d stride,
const dim2d padding
const shape2d poolingSize,
const shape2d stride,
const shape2d padding
);
} // namespace CUDANet::Kernels