max_pool3d_backward_f32

void max_pool3d_backward_f32(const float * g, const int * argmax, float * dx, int B, int C, int D, int H, int W, int OD, int OH, int OW)

Single-precision 3-D max-pool backward — scatters each upstream gradient onto the input voxel that won its pool window (per the forward argmax).

dx is accumulated and must be zeroed by the caller.

gconst float*

Upstream gradient of shape (B, C, OD, OH, OW).

argmaxconst int32_t*

Per-output winning flat 3-D index from the matching forward call.

dxfloat*

Output gradient of shape (B, C, D, H, W); accumulated.

B, C, D, H, W, OD, OH, OWint

Layout extents.