Commit Graph

6 Commits (a7a4f053dacd028469c0fd9a2a9e6e54eb3fa55d)

Author SHA1 Message Date
chengduo e1761709f8
Set the right shape of selected_rows (#13723)
7 years ago
Yibing Liu a523b6f49f Add python api 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 4760f2851e Add the argsort operator
7 years ago