提交 e6014b60 authored 作者: Reyhane Askari's avatar Reyhane Askari

minor fix

上级 a70a1e51
...@@ -1332,7 +1332,7 @@ def test_argmax_pushdown(): ...@@ -1332,7 +1332,7 @@ def test_argmax_pushdown():
# print 'AFTER' # print 'AFTER'
# for node in fgraph.toposort(): # for node in fgraph.toposort():
# print node.op # print node.op
assert len(fgraph.toposort()) == 1 # an output_guard is second assert len(fgraph.toposort()) == 1
assert fgraph.toposort()[0].op == tensor.basic._argmax assert fgraph.toposort()[0].op == tensor.basic._argmax
assert check_stack_trace( assert check_stack_trace(
fgraph, ops_to_check=tensor.basic._argmax) fgraph, ops_to_check=tensor.basic._argmax)
...@@ -1357,7 +1357,7 @@ def test_argmax_pushdown(): ...@@ -1357,7 +1357,7 @@ def test_argmax_pushdown():
# print 'AFTER' # print 'AFTER'
# for node in fgraph.toposort(): # for node in fgraph.toposort():
# print node.op # print node.op
assert len(fgraph.toposort()) == 3 # an output_guard is second assert len(fgraph.toposort()) == 3
assert isinstance(fgraph.toposort()[0].op, tensor.Elemwise) assert isinstance(fgraph.toposort()[0].op, tensor.Elemwise)
assert isinstance(fgraph.toposort()[1].op, Softmax) assert isinstance(fgraph.toposort()[1].op, Softmax)
assert isinstance(fgraph.toposort()[2].op, tensor.CAReduce) assert isinstance(fgraph.toposort()[2].op, tensor.CAReduce)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论