Commit Graph

10 Commits (252f1e4a349bd06557a13c64807ded268bbeac46)

Author SHA1 Message Date
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
dangqingqing f5e367655e Use G++ to compile some cu operators.
7 years ago
Yu Yang 3a5693e0a8 Add Skeleton of Double support
8 years ago
xzl 0cd9b8c0aa modify the input\output name to X\Out
8 years ago
xzl 9de45e113a fixed bug when dims.size == 1, modify the variable naming, add judgement when input_grad is null
8 years ago
xzl 5ede6fd434 delete cuda impl, complete comments, modify variable naming
8 years ago
xzl 61c7930603 delete useless header file
8 years ago
xzl 55991822a0 modify GetAttr to Attr
8 years ago
xzl d6651b9b8e fixed bug of the gpu impl
8 years ago
xzl 17b4b980e7 add the transpose op
8 years ago