Merge branch 'pr388'
Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/c8df172c Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/c8df172c Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/c8df172c Branch: refs/heads/master Commit: c8df172c71941913d0aaea0d70782a4d649ae0be Parents: 9b06c29 f9e7caa Author: Wang Wei <[email protected]> Authored: Sat Jun 30 21:30:20 2018 +0800 Committer: Wang Wei <[email protected]> Committed: Sat Jun 30 21:30:20 2018 +0800 ---------------------------------------------------------------------- include/singa/core/tensor.h | 6 +- src/core/tensor/tensor.cc | 151 +++++++--- src/core/tensor/tensor_math.h | 8 + src/core/tensor/tensor_math_cpp.h | 62 ++-- src/core/tensor/tensor_math_cuda.h | 502 ++++++++++++++++++++++++-------- 5 files changed, 543 insertions(+), 186 deletions(-) ----------------------------------------------------------------------
