Commit Graph

8 Commits (7d653c41cd2c0de2c301097df9e0dc8f08e5c59c)

Author SHA1 Message Date
Yu Yang c532b96741 Remove template parameter for Tensor methods (#4937)
7 years ago
QI JUN 2603cb7e86 Unify CUDA stream in Tensor CopyFrom interface (#4692)
7 years ago
Yu Yang 3a5693e0a8 Add Skeleton of Double support
8 years ago
Helin Wang 25be0ede76 fix cpplint error
8 years ago
Yibing Liu 0289a0091f follow comments to cleanup code
8 years ago
Yibing Liu dd64349a92 refine reshape operator
8 years ago
Yibing Liu 899c7d6b35 pass unit test
8 years ago
Yibing Liu 12eaa22ad2 add reshape operator
8 years ago