提交 f5224180 authored 作者: Cesar Laurent's avatar Cesar Laurent

Corrected infer_shape to work with shared variables (fix test_infer_shape).

上级 e326ee52
...@@ -187,9 +187,9 @@ class Pool(OpenMPOp): ...@@ -187,9 +187,9 @@ class Pool(OpenMPOp):
r = tensor.extract_constant(r) r = tensor.extract_constant(r)
c = tensor.extract_constant(c) c = tensor.extract_constant(c)
if padding[0]: if padding[0]:
r += padding[0] * 2 r = r + padding[0] * 2
if padding[1]: if padding[1]:
c += padding[1] * 2 c = c + padding[1] * 2
if ignore_border: if ignore_border:
if ds[0] == st[0]: if ds[0] == st[0]:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论