SINGA-190 - Add prelu layer and flatten layer Merge PR#162 into dev
Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/26df5ac0 Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/26df5ac0 Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/26df5ac0 Branch: refs/heads/master Commit: 26df5ac03326576cebcca516da3b27ba1fd0dbd8 Parents: 58be3f8 6d69047 Author: Wei Wang <[email protected]> Authored: Sun Jun 12 22:33:02 2016 +0800 Committer: Wei Wang <[email protected]> Committed: Sun Jun 12 22:33:02 2016 +0800 ---------------------------------------------------------------------- include/singa/core/tensor.h | 396 ++++++++--------- src/core/tensor/math_kernel.cu | 656 +++++++++++++++------------- src/core/tensor/math_kernel.h | 93 ++-- src/core/tensor/tensor.cc | 702 ++++++++++++++++-------------- src/core/tensor/tensor_math.h | 393 +++++++++-------- src/core/tensor/tensor_math_cpp.h | 585 +++++++++++++++++++------ src/core/tensor/tensor_math_cuda.h | 412 ++++++++++++++---- src/model/layer/cudnn_convolution.cc | 180 ++++---- test/singa/test_cudnn_convolution.cc | 181 ++++++++ test/singa/test_tensor_math.cc | 295 ++++++++++++- 10 files changed, 2470 insertions(+), 1423 deletions(-) ----------------------------------------------------------------------
