提交 2aec1bab authored 作者: Frederic Bastien's avatar Frederic Bastien

rename opt and add comment

上级 b4aefd7d
......@@ -23,7 +23,7 @@ theano.compile.optdb.register('local_inplace_remove0',
@gof.local_optimizer([csm_properties])
def skip_pack_csc01(node):
def local_csm_properties_csm(node):
"""if we find csm_properties(CSM(*args)), then we can replace that with the
*args directly"""
if node.op == csm_properties:
......@@ -36,4 +36,4 @@ def skip_pack_csc01(node):
return ret_var
return False
register_specialize(skip_pack_csc01)
register_specialize(local_csm_properties_csm)
......@@ -1188,7 +1188,8 @@ def test_remove0():
f = theano.function([theano.In(x, borrow=True, mutable=True)],
Remove0()(x))
# assert optimization is applied in modes with optimization
# assert optimization local_inplace_remove0 is applied in
# modes with optimization
if theano.config.mode not in ['FAST_COMPILE']:
# list of apply nodes in the optimized graph.
nodes = f.maker.env.toposort()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论