提交 013fe629 authored 作者: f0k's avatar f0k

added missing `host_from_gpu` import to `cuda.opt_util`

上级 c0217dd1
...@@ -8,7 +8,7 @@ from theano.gof import local_optimizer ...@@ -8,7 +8,7 @@ from theano.gof import local_optimizer
from theano.tensor import DimShuffle from theano.tensor import DimShuffle
from theano.sandbox.cuda.basic_ops import ( from theano.sandbox.cuda.basic_ops import (
GpuFromHost, HostFromGpu, GpuDimShuffle, GpuElemwise) GpuFromHost, HostFromGpu, host_from_gpu, GpuDimShuffle, GpuElemwise)
def grab_cpu_scalar(v, nd): def grab_cpu_scalar(v, nd):
if v.owner is not None: if v.owner is not None:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论