提交 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():
rr = conv * lr
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,
dnn.GpuDnnConv)
assert not isinstance(f.maker.fgraph.outputs[1].owner.inputs[0].owner.op,
dnn.GpuDnnConv)
convs = [n for n in f.maker.fgraph.toposort()
if isinstance(n.op, dnn.GpuDnnConv)]
assert len(convs) == 1
def test_dnn_conv_merge_broad():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论