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

Move blocksparse to tensor.nnet

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