提交 f90d326e authored 作者: AdeB's avatar AdeB

Move blocksparse to tensor.nnet

上级 619b9859
......@@ -4,7 +4,7 @@ from nose.plugins.skip import SkipTest
import theano
from theano import tensor
import theano.tests.unittest_tools as utt
import theano.sandbox.tests.test_blocksparse
import theano.tensor.nnet.tests.test_blocksparse
import theano.sandbox.cuda as cuda_ndarray
from theano.sandbox.cuda.blocksparse import (GpuSparseBlockOuter,
......
......@@ -29,7 +29,7 @@ from theano.sandbox.cuda import basic_ops
from theano.sandbox.cuda.type import CudaNdarrayType
from theano.scalar.basic_scipy import erfinv
from theano.sandbox.blocksparse import sparse_block_dot
from theano.tensor.nnet.blocksparse import sparse_block_dot
from theano.sandbox.cuda.blocksparse import GpuSparseBlockGemv, GpuSparseBlockOuter
......
from .nnet import *
from . import opt
from .conv import conv2d, ConvOp
from .Conv3D import *
from .ConvGrad3D import *
......
......@@ -29,7 +29,7 @@ from theano.gof import Apply
from theano.tensor.nnet.sigm import sigmoid, softplus
from theano.gradient import DisconnectedType
from theano.gradient import grad_not_implemented
from theano.sandbox.blocksparse import sparse_block_dot
from theano.tensor.nnet.blocksparse import sparse_block_dot
from theano.tensor.type import values_eq_approx_remove_nan
......
"""
Optimizations addressing the ops in sandbox root directory
Optimizations addressing the ops in nnet root directory
"""
from theano import compile # to register the optimizer built by this file
from theano import gof
from theano.sandbox.blocksparse import (
from theano.tensor.nnet.blocksparse import (
SparseBlockGemv,
SparseBlockOuter,
sparse_block_gemv_inplace,
......
......@@ -10,7 +10,7 @@ import theano
from theano import tensor
import theano.tests.unittest_tools as utt
from theano.sandbox.blocksparse import sparse_block_dot, \
from theano.tensor.nnet.blocksparse import sparse_block_dot, \
sparse_block_gemv, sparse_block_outer
......
import theano
from theano import tensor
from theano.sandbox.blocksparse import sparse_block_dot
from theano.tensor.nnet.blocksparse import sparse_block_dot
def test_blocksparse_inplace_gemv_opt():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论