std1.0.0-beta-7 SafeC docs

std :: ml::gpu_rocm

Function std::rocm_half_avgpool2d

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

int rocm_half_avgpool2d(
    struct RocmHalfBuf x,
    struct RocmHalfBuf y,
    unsigned long batch,
    unsigned long channels,
    unsigned long height,
    unsigned long width,
    unsigned long kernel,
    unsigned long stride,
)