提交 0838ae4d authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #2659 from abergeron/tests

Add a test for all possibilities of border_mode, fix #2600.
......@@ -466,6 +466,19 @@ class TestDnnInferShapes(utt.InferShapeTester):
)
# this has been a problem in the past
def test_dnn_conv_border_mode():
if not cuda.dnn.dnn_available():
raise SkipTest(cuda.dnn.dnn_available.msg)
img = T.ftensor4()
kern = T.ftensor4()
dnn.dnn_conv(img, kern, border_mode=1)
dnn.dnn_conv(img, kern, border_mode=(2, 3))
dnn.dnn_conv(img, kern, border_mode='full')
dnn.dnn_conv(img, kern, border_mode='valid')
def test_dnn_conv_merge():
if not cuda.dnn.dnn_available():
raise SkipTest(cuda.dnn.dnn_available.msg)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论