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

commented import of gpu code in theano.tensor. They should not be there.

上级 45cf0c51
import theano import theano
from theano.tensor import basic as T from theano.tensor import basic as T
import numpy as N import numpy as N
from theano.sandbox.cuda import cuda_available, cuda_enabled
#from util import strutil #from util import strutil
from theano import printing from theano import printing
from theano.tensor.blas_headers import blas_header_text from theano.tensor.blas_headers import blas_header_text
...@@ -570,6 +569,6 @@ def computeH(V,W,b,d): ...@@ -570,6 +569,6 @@ def computeH(V,W,b,d):
from . import ConvGrad3D from . import ConvGrad3D
from . import ConvTransp3D from . import ConvTransp3D
if cuda_available: #from theano.sandbox.cuda import cuda_available, cuda_enabled
from . import GpuConv3D #if cuda_available:
# from . import GpuConv3D
import theano import theano
from theano.tensor import basic as T from theano.tensor import basic as T
from theano.sandbox.cuda import cuda_available, cuda_enabled
#from util import strutil #from util import strutil
import numpy as N import numpy as N
...@@ -269,6 +268,7 @@ convGrad3D = ConvGrad3D() ...@@ -269,6 +268,7 @@ convGrad3D = ConvGrad3D()
from Conv3D import conv3D from Conv3D import conv3D
from ConvTransp3D import convTransp3D from ConvTransp3D import convTransp3D
if cuda_available: #from theano.sandbox.cuda import cuda_available, cuda_enabled
from theano.sandbox.cuda import CudaNdarrayType, float32_shared_constructor #if cuda_available:
from . import GpuConvGrad3D # from theano.sandbox.cuda import CudaNdarrayType, float32_shared_constructor
# from . import GpuConvGrad3D
...@@ -2,7 +2,6 @@ import numpy as N ...@@ -2,7 +2,6 @@ import numpy as N
from theano.tensor import basic as T from theano.tensor import basic as T
#from util import strutil #from util import strutil
import theano import theano
from theano.sandbox.cuda import cuda_available, cuda_enabled
class ConvTransp3D(theano.Op): class ConvTransp3D(theano.Op):
""" "Transpose" of Conv3D (Conv3D implements multiplication by an implicitly defined matrix W. This implements multiplication by its transpose) """ """ "Transpose" of Conv3D (Conv3D implements multiplication by an implicitly defined matrix W. This implements multiplication by its transpose) """
...@@ -405,5 +404,6 @@ def computeR(W,b,d,H,Rshape = None): ...@@ -405,5 +404,6 @@ def computeR(W,b,d,H,Rshape = None):
from Conv3D import conv3D from Conv3D import conv3D
from ConvGrad3D import convGrad3D from ConvGrad3D import convGrad3D
if cuda_available: #from theano.sandbox.cuda import cuda_available, cuda_enabled
import GpuConvTransp3D #if cuda_available:
# import GpuConvTransp3D
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论