提交 8d45feab authored 作者: --global's avatar --global

Add comment in test_log_softmax_opt

上级 8df74d4d
......@@ -369,6 +369,7 @@ def test_log_softmax_opt():
dnn_softmax_nodes = [n for n in f.maker.fgraph.toposort() if
isinstance(n.op, cuda.dnn.GpuDnnSoftmax)]
# Ensure that the optimization has been applied
assert len(dnn_softmax_nodes) == 1
assert dnn_softmax_nodes[0].op.algo == "log"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论