提交 5af163f7 authored 作者: Frederic Bastien's avatar Frederic Bastien

Simplify graph build

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