提交 3372bb22 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make test_neighbours pass.

上级 8e45b3fd
...@@ -460,7 +460,7 @@ class GpuImages2Neibs(GpuKernelBase, Images2Neibs, Op): ...@@ -460,7 +460,7 @@ class GpuImages2Neibs(GpuKernelBase, Images2Neibs, Op):
@op_lifter([Images2Neibs]) @op_lifter([Images2Neibs])
def use_gpu_images2neibs(node): def use_gpu_images2neibs(node, context_name):
if node.op.mode in ['valid', 'ignore_borders', 'wrap_centered']: if node.op.mode in ['valid', 'ignore_borders', 'wrap_centered']:
return GpuImages2Neibs(node.op.mode) return GpuImages2Neibs(node.op.mode)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论