提交 73c84b5b authored 作者: Frederic's avatar Frederic

small code cleanup

上级 04f79550
import copy
import os
import theano
from theano import Apply, tensor
from theano.gof.type import CDataType
from theano.compat import PY3
from theano.compat.six import StringIO
from theano.sandbox.cuda.type import CudaNdarrayType
from theano.sandbox.cuda import (GpuOp, cuda_available, active_device_number,
device_properties)
......@@ -926,8 +924,7 @@ err%(name)s = cudnnSoftmaxForward(
# We need this since other stuff from opt is not importable.
if cuda_available:
from theano.sandbox.cuda.opt import (local_optimizer, gpu_contiguous,
gpu_optimizer)
from theano.sandbox.cuda.opt import local_optimizer, gpu_optimizer
@local_optimizer([GpuConv])
def local_conv_dnn(node):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论