提交 6dfdace2 authored 作者: Frederic's avatar Frederic

work around gh-3444

上级 da3c8070
...@@ -1457,6 +1457,8 @@ class GpuDnnPool(DnnBase): ...@@ -1457,6 +1457,8 @@ class GpuDnnPool(DnnBase):
return Apply(self, [img, desc], [img.type()]) return Apply(self, [img, desc], [img.type()])
def infer_shape(self, node, shape): def infer_shape(self, node, shape):
if not node.inputs[1].owner:
raise theano.tensor.ShapeError()
desc = node.inputs[1].owner.op desc = node.inputs[1].owner.op
nd = desc.get_ndim() nd = desc.get_ndim()
w = desc.ws w = desc.ws
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论