提交 6a2639ad authored 作者: James Bergstra's avatar James Bergstra

Downsample.make_node checks arg type

上级 9b3168ae
...@@ -204,6 +204,10 @@ class GpuDownsampleFactorMax(Op): ...@@ -204,6 +204,10 @@ class GpuDownsampleFactorMax(Op):
return '%s{%s,%s}' % (self.__class__.__name__, self.ds, self.ignore_border) return '%s{%s,%s}' % (self.__class__.__name__, self.ds, self.ignore_border)
def make_node(self, x): def make_node(self, x):
if not isinstance(x.type, CudaNdarrayType):
raise TypeError()
if not x.type.ndim == 4:
raise TypeError()
return Apply(self, [x], [x.type()]) return Apply(self, [x], [x.type()])
#def perform(self, node, input_storage, output_storage): #def perform(self, node, input_storage, output_storage):
#raise NotImplementedError('only C is implemented') #raise NotImplementedError('only C is implemented')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论