Commit Graph

9 Commits (f1a9efcac5743e05fc4e1dbc53c08f69ec19d5a8)

Author SHA1 Message Date
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
Yu Yang 3a5693e0a8 Add Skeleton of Double support
8 years ago
yangyaming 4e3ba65f19 Refine doc.
8 years ago
yangyaming 09a13f6c49 Using LoDTensor and adapt to new unittest.
8 years ago
yangyaming 076dcb9b45 Simpify the initialization for weights.
8 years ago
yangyaming 53ab7e78b1 Adapt new interface.
8 years ago
yangyaming f5807670a7 Fix typos and use HOSTDEVICE instead.
8 years ago
yangyaming b7776e66d6 Fix dimension bugs.
8 years ago
yangyaming 2763f3e32f Complete smooth_l1_loss_op.
8 years ago