提交 dabc3dd5 authored 作者: Frederic Bastien's avatar Frederic Bastien

Register in fast_compile an opt that move to the GPU.

上级 b37658f3
...@@ -1449,7 +1449,7 @@ from theano.sandbox.gpuarray.opt import (register_opt as register_gpua, ...@@ -1449,7 +1449,7 @@ from theano.sandbox.gpuarray.opt import (register_opt as register_gpua,
host_from_gpu as host_from_gpua) host_from_gpu as host_from_gpua)
@register_gpua() @register_gpua('fast_compile')
@local_optimizer([mrg_uniform]) @local_optimizer([mrg_uniform])
def local_gpua_mrg(node): def local_gpua_mrg(node):
if (type(node.op) == mrg_uniform and if (type(node.op) == mrg_uniform and
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论