std :: ml::gpu_rocm
Function std::rocm_half_maxpool2d
int rocm_half_maxpool2d(
struct RocmHalfBuf x,
struct RocmHalfBuf y,
unsigned long batch,
unsigned long channels,
unsigned long height,
unsigned long width,
unsigned long kernel,
unsigned long stride,
)16-bit max / avg pooling on NCHW (x: [batch, channels, height, width]), kernel x kernel window, given stride, no padding: the same windows as gpu_miopen.h's rocm_maxpool2d. fp32 accumulation, one rounding; max pooling copies the largest value's bits.