提交 12e9ba70 authored 作者: Frederic Bastien's avatar Frederic Bastien

test GpuDownsampleFactorMaxGrad with not square downsample.

上级 2bf5a874
...@@ -164,7 +164,7 @@ def test_downsample(): ...@@ -164,7 +164,7 @@ def test_downsample():
(30, 2, 24, 24), (30, 2, 24, 24),
(30, 6, 24, 24), (30, 6, 24, 24),
(10, 10, 10, 11)]: (10, 10, 10, 11)]:
for ds in (2, 2), (1,1): for ds in (2, 2), (3,2), (1,1):
if ds[0] > shp[2]: continue if ds[0] > shp[2]: continue
if ds[1] > shp[3]: continue if ds[1] > shp[3]: continue
for ignore_border in (True, False): for ignore_border in (True, False):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论