max_pool2d_forward_f64

void max_pool2d_forward_f64(const double * x, double * y, int * argmax, int B, int C, int H, int W, int KH, int KW, int OH, int OW, int stride_h, int stride_w, int pad_h, int pad_w)

source edit

Double-precision counterpart to max_pool2d_forward_f32.

Parameters

xconst double*

Input tensor of shape (B, C, H, W).

ydouble*

Output tensor of shape (B, C, OH, OW).

argmaxint32_t*

Per-output winning h*W + w indices.

B, C, H, W, KH, KW, OH, OW, stride_h, stride_w, pad_h, pad_wint

Layout and window parameters; see max_pool2d_forward_f32.