提交 6dacc650 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Also register the optimizations in the specialize phase.

上级 23817993
...@@ -308,7 +308,7 @@ def local_dimshuffle_no_inplace_at_canonicalize(node): ...@@ -308,7 +308,7 @@ def local_dimshuffle_no_inplace_at_canonicalize(node):
###################### ######################
@register_canonicalize @register_canonicalize
#@register_specialize @register_specialize
@gof.local_optimizer([T.TensorFromScalar]) @gof.local_optimizer([T.TensorFromScalar])
def local_tensor_scalar_tensor(node): def local_tensor_scalar_tensor(node):
'''tensor_from_scalar(scalar_from_tensor(x)) -> x''' '''tensor_from_scalar(scalar_from_tensor(x)) -> x'''
...@@ -319,7 +319,7 @@ def local_tensor_scalar_tensor(node): ...@@ -319,7 +319,7 @@ def local_tensor_scalar_tensor(node):
return [t] return [t]
@register_canonicalize @register_canonicalize
#@register_specialize @register_specialize
@gof.local_optimizer([T.ScalarFromTensor]) @gof.local_optimizer([T.ScalarFromTensor])
def local_scalar_tensor_scalar(node): def local_scalar_tensor_scalar(node):
'''scalar_from_tensor(tensor_from_scalar(x)) -> x''' '''scalar_from_tensor(tensor_from_scalar(x)) -> x'''
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论