提交 b35d1bb6 authored 作者: Frederic's avatar Frederic

rename opt

上级 0bbdb68c
......@@ -1426,9 +1426,11 @@ optdb.register('crossentropy_to_crossentropy_with_softmax',
'fast_run', 'xent', 'fast_compile_gpu')
@opt.register_specialize('fast_compile_gpu')
@opt.register_specialize(
'fast_compile_gpu',
'local_crossentropy_to_crossentropy_with_softmax_grad') # old name
@gof.local_optimizer([softmax_grad])
def local_crossentropy_to_crossentropy_with_softmax_grad(node):
def local_softmax_grad_to_crossentropy_with_softmax_grad(node):
if node.op == softmax_grad:
g_coding_dist, coding_dist = node.inputs
if (g_coding_dist.owner and
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论