提交 0c17be82 authored 作者: carriepl's avatar carriepl

Add tests for border_mode half

上级 3ee95b87
...@@ -389,7 +389,7 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -389,7 +389,7 @@ class TestDnnInferShapes(utt.InferShapeTester):
) )
for params in product( for params in product(
['valid', 'full'], ['valid', 'full', 'half'],
[(1, 1), (2, 2)], [(1, 1), (2, 2)],
['conv', 'cross'] ['conv', 'cross']
): ):
...@@ -427,7 +427,7 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -427,7 +427,7 @@ class TestDnnInferShapes(utt.InferShapeTester):
) )
for params in product( for params in product(
['valid', 'full'], ['valid', 'full', 'half'],
[(1, 1)], # strides besides (1, 1) [(1, 1)], # strides besides (1, 1)
['conv', 'cross'] ['conv', 'cross']
): ):
...@@ -590,6 +590,7 @@ def test_dnn_conv_border_mode(): ...@@ -590,6 +590,7 @@ def test_dnn_conv_border_mode():
dnn.dnn_conv(img, kern, border_mode=(2, 3)) dnn.dnn_conv(img, kern, border_mode=(2, 3))
dnn.dnn_conv(img, kern, border_mode='full') dnn.dnn_conv(img, kern, border_mode='full')
dnn.dnn_conv(img, kern, border_mode='valid') dnn.dnn_conv(img, kern, border_mode='valid')
dnn.dnn_conv(img, kern, border_mode='half')
def test_dnn_conv_alpha_output_merge(): def test_dnn_conv_alpha_output_merge():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论