Commit Graph

10 Commits (42d67dacb652f1d16406cbcdd36acd429c1e9c7b)

Author SHA1 Message Date
FlyingQianMM 443a713c9e
add backward gradient computation for op argsort (#22203)
6 years ago
Aurelius84 5cb2c74127
add register op_data_type of pad/expand_as et.al (#21718)
6 years ago
zhaoyuchen2018 b16274556a
Add dscending for argsort (#21400)
6 years ago
zhaoyuchen2018 08c19c585d
Improve argsort performance. (#21267)
6 years ago
Tao Luo 02270b3eb1
remove unused assert.h (#19529)
6 years ago
Yibing Liu 9386ac0a40 Enhance cuda code & unittest for argsort_op
7 years ago
Yibing Liu 92cfa2be3a Avoid using dynamic array in cuda kernel
7 years ago
Yibing Liu 94e72ea6e7 Support more negative axes in argsort_op
7 years ago
Yibing Liu 42645ff779 Compute target index on gpu
7 years ago
Yibing Liu 6ee22c4f71 Add gpu kernel for argsort op
7 years ago