std1.0.0-beta-7 SafeC docs

std :: ml::gpu_cudnn

Function std::cuda_maxpool2d

#include <std/ml/gpu_cudnn.h> source ยท definition

int cuda_maxpool2d(
    struct CudaBuf x,
    struct CudaBuf y,
    unsigned long batch,
    unsigned long channels,
    unsigned long height,
    unsigned long width,
    unsigned long kernel,
    unsigned long stride,
)

kernel x kernel window, given stride, no padding (FeatureMap::maxpool2d / FeatureMap::avgpool2d). x: [batch, channels, height, width].