提交 f7c01d53 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add fast_compile for the lifter.

上级 38199e38
......@@ -485,7 +485,7 @@ def local_gpua_multinomial(op, context_name, inputs, outputs):
gpu_op(p, u))
@register_opt()
@register_opt('fast_compile')
@op_lifter([theano.sandbox.multinomial.MultinomialWOReplacementFromUniform])
@register_opt2([theano.sandbox.multinomial.MultinomialWOReplacementFromUniform], 'fast_compile')
def local_gpua_multinomial_wor(op, context_name, inputs, outputs):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论