提交 50a13568 authored 作者: Frederic's avatar Frederic

comment fix following code review.

上级 f85d7e34
......@@ -426,6 +426,7 @@ def dimshuffle_as_view(node):
new_op = DimShuffle(op.input_broadcastable, op.new_order, inplace=True)
return [new_op(*node.inputs)]
#Step 60 is the inplace optimization stage.
compile.optdb.register('dimshuffle_as_view',
TopoOptimizer(dimshuffle_as_view,
failure_callback=TopoOptimizer.warn_inplace), 60,
......@@ -3145,7 +3146,7 @@ def local_cut_useless_reduce(node):
#Enabling this optimization at canonicalization step break this test:
#theano/tensor/tests/test_opt.py:T_local_sum.test_local_sum_broadcast_some_0
#theano/tensor/tests/test_opt.py:T_local_reduce.test_local_reduce_broadcast_some_0
# see gh-790 issue.
#
#@register_canonicalize
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论