提交 9c960345 authored 作者: Alexander Matyasko's avatar Alexander Matyasko

Fix typo in average pooling 3d kernel

上级 b0c7744e
......@@ -88,7 +88,7 @@ KERNEL void ave_pool3d_grad_kernel(const ga_size nthreads,
const ga_size dstart = pd * stride_d - pad_d;
const ga_size hstart = ph * stride_h - pad_h;
const ga_size wstart = pw * stride_w - pad_w;
const ga_size dend = min(dstart + kernel_h, depth + pad_d);
const ga_size dend = min(dstart + kernel_d, depth + pad_d);
const ga_size hend = min(hstart + kernel_h, height + pad_h);
const ga_size wend = min(wstart + kernel_w, width + pad_w);
const ga_size pool_size = (dend - dstart) * (hend - hstart) * (wend - wstart);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论