提交 e97223be authored 作者: Gabe Schwartz's avatar Gabe Schwartz

Updated cudnn tests to always handle dilation.

上级 6217e848
...@@ -641,7 +641,7 @@ class GpuDnnConv(DnnBase): ...@@ -641,7 +641,7 @@ class GpuDnnConv(DnnBase):
return [[1], [1], [1], [0], [1], [1]] return [[1], [1], [1], [0], [1], [1]]
@staticmethod @staticmethod
def get_out_shape(ishape, kshape, border_mode, subsample): def get_out_shape(ishape, kshape, border_mode, subsample, dilation):
""" """
This function computes the output shape for a convolution with This function computes the output shape for a convolution with
the specified parameters. `ishape` and `kshape` can be symbolic the specified parameters. `ishape` and `kshape` can be symbolic
...@@ -660,7 +660,8 @@ class GpuDnnConv(DnnBase): ...@@ -660,7 +660,8 @@ class GpuDnnConv(DnnBase):
ishape, ishape,
kshape, kshape,
border_mode, border_mode,
subsample) subsample,
dilation)
def infer_shape(self, node, shape): def infer_shape(self, node, shape):
return [shape[2]] return [shape[2]]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论