提交 95d9c1a5 authored 作者: jiakai's avatar jiakai

test other border_mode

上级 95ee0448
...@@ -870,10 +870,13 @@ class TestConvWithPadding(object): ...@@ -870,10 +870,13 @@ class TestConvWithPadding(object):
assert_allclose(cpuval, gpuval, rtol=1e-5, atol=1e-5) assert_allclose(cpuval, gpuval, rtol=1e-5, atol=1e-5)
def test_numeric_value(self): def test_numeric_value(self):
shape_param = [ params = [
((5, 10, 4, 4), (12, 10, 4, 4), (2, 1)) ((5, 10, 4, 4), (12, 10, 4, 4), (2, 1)),
((5, 10, 8, 8), (12, 10, 4, 4), 3),
((5, 10, 6, 8), (12, 10, 3, 4), 'full'),
((5, 10, 9, 6), (12, 10, 9, 4), 'valid')
] ]
for img_shape, kern_shape, padding in shape_param: for img_shape, kern_shape, padding in params:
yield (self._run_onecase, img_shape, kern_shape, padding) yield (self._run_onecase, img_shape, kern_shape, padding)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论