提交 5a62e6f6 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Restore patternbroadcast.

上级 b722302f
...@@ -842,6 +842,10 @@ def local_gpu_conv(node, context_name): ...@@ -842,6 +842,10 @@ def local_gpu_conv(node, context_name):
GpuFromHost(context_name)(kern)) GpuFromHost(context_name)(kern))
assert isinstance(out.type, GpuArrayType) assert isinstance(out.type, GpuArrayType)
out.values_eq_approx = values_eq_approx out.values_eq_approx = values_eq_approx
# Make sure to keep the broadcastable pattern of the original
# convolution even if we might gain or lose some due to different
# information at the node level.
out = tensor.patternbroadcast(out, node.outputs[0].broadcastable)
return [out] return [out]
# Register this here so that it goes after 'local_gpu_conv' # Register this here so that it goes after 'local_gpu_conv'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论