提交 999f6399 authored 作者: Frederic's avatar Frederic

Use the good type.

上级 5682d3f8
...@@ -370,7 +370,7 @@ def local_gpu_conv(node): ...@@ -370,7 +370,7 @@ def local_gpu_conv(node):
if a.shape[-1] * a.shape[-2] > 100: if a.shape[-1] * a.shape[-2] > 100:
#For float32 the default atol is 1e-5 #For float32 the default atol is 1e-5
atol = 3e-5 atol = 3e-5
return CudaNdarrayType.values_eq_approx(a, b, atol=atol) return GpuArrayType.values_eq_approx(a, b, atol=atol)
img, kern = node.inputs img, kern = node.inputs
gpu_conv = GpuConvOp_from_ConvOp(node.op) gpu_conv = GpuConvOp_from_ConvOp(node.op)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论