提交 c3553747 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make the test easier to read/understand.

上级 e0bf5038
...@@ -680,10 +680,9 @@ def test_dnn_conv_merge_mouts(): ...@@ -680,10 +680,9 @@ def test_dnn_conv_merge_mouts():
rr = conv * lr rr = conv * lr
f = theano.function([img, kern, out], [fr, rr], mode=mode_with_gpu) f = theano.function([img, kern, out], [fr, rr], mode=mode_with_gpu)
assert not isinstance(f.maker.fgraph.outputs[0].owner.inputs[0].owner.op, convs = [n for n in f.maker.fgraph.toposort()
dnn.GpuDnnConv) if isinstance(n.op, dnn.GpuDnnConv)]
assert not isinstance(f.maker.fgraph.outputs[1].owner.inputs[0].owner.op, assert len(convs) == 1
dnn.GpuDnnConv)
def test_dnn_conv_merge_broad(): def test_dnn_conv_merge_broad():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论