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

fix error detection.

上级 73c84b5b
...@@ -119,9 +119,9 @@ class GpuDnnConvDesc(GpuOp): ...@@ -119,9 +119,9 @@ class GpuDnnConvDesc(GpuOp):
self.conv_mode = conv_mode self.conv_mode = conv_mode
def make_node(self, img_shape, kern_shape): def make_node(self, img_shape, kern_shape):
if img_shape.type.ndim != 1 and img_shape.type.dtype != numpy.int64: if img_shape.type.ndim != 1 or img_shape.type.dtype != 'int64':
raise TypeError('img must be 1D shape tensor') raise TypeError('img must be 1D shape tensor')
if kern_shape.type.ndim != 1 and kern_shape.type.dtype != numpy.int64: if kern_shape.type.ndim != 1 or kern_shape.type.dtype != 'int64':
raise TypeError('kern must be 1D shape tensor') raise TypeError('kern must be 1D shape tensor')
return Apply(self, [img_shape, kern_shape], return Apply(self, [img_shape, kern_shape],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论