提交 561ac97c authored 作者: Frederic's avatar Frederic

Speed up DownsampleFactorMaxGrad.perform

上级 2b42af36
...@@ -309,8 +309,9 @@ class DownsampleFactorMaxGrad(Op): ...@@ -309,8 +309,9 @@ class DownsampleFactorMaxGrad(Op):
zj = j // ds1 zj = j // ds1
if (maxout[n, k, zi, zj] == x[n, k, i, j]): if (maxout[n, k, zi, zj] == x[n, k, i, j]):
gx[n, k, i, j] = gz[n, k, zi, zj] gx[n, k, i, j] = gz[n, k, zi, zj]
else: # Not needed as it is allocated with zeros
gx[n, k, i, j] = 0 # else:
# gx[n, k, i, j] = 0
gx_stg[0] = gx gx_stg[0] = gx
def infer_shape(self, node, in_shapes): def infer_shape(self, node, in_shapes):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论