提交 994c8d35 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Register the logsoftmax opt in fast_compile.

上级 b1c5a67e
...@@ -736,7 +736,7 @@ class LogSoftmax(gof.Op): ...@@ -736,7 +736,7 @@ class LogSoftmax(gof.Op):
logsoftmax_op = LogSoftmax() logsoftmax_op = LogSoftmax()
@opt.register_specialize('stabilize') @opt.register_specialize('stabilize', 'fast_compile')
@gof.local_optimizer([tensor.Elemwise]) @gof.local_optimizer([tensor.Elemwise])
def local_logsoftmax(node): def local_logsoftmax(node):
""" """
...@@ -757,7 +757,7 @@ def local_logsoftmax(node): ...@@ -757,7 +757,7 @@ def local_logsoftmax(node):
pass pass
@opt.register_specialize('stabilize') @opt.register_specialize('stabilize', 'fast_compile')
@gof.local_optimizer([SoftmaxGrad]) @gof.local_optimizer([SoftmaxGrad])
def local_logsoftmax_grad(node): def local_logsoftmax_grad(node):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论