提交 d708a276 authored 作者: Frederic's avatar Frederic

[CRASH] Fix crash with partial shape info in conv2d on the CPU in FAST_COMPILE.

This was discovered by the daily buildbot.
上级 f0a17009
......@@ -695,6 +695,12 @@ class ConvOp(OpenMPOp):
kshp_logical = self.kshp_logical
if kshp_logical is None:
kshp_logical = kshp
else:
if kshp_logical[0] is None:
kshp_logical = (kshp[0], kshp_logical[1])
if kshp_logical[1] is None:
kshp_logical = (kshp_logical[0], kshp[1])
if numpy.any([x is None for x in kshp_logical]):
kshp = tuple(filtersflipped.shape[2:])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论