Commit Graph

8 Commits (8c19d7aa2f89a38b3a68e53c73d88af16a3de8ce)

Author SHA1 Message Date
FlyingQianMM 79d712346f
Correct CPU gradients of the argsort op (#22739)
5 years ago
FlyingQianMM 443a713c9e
add backward gradient computation for op argsort (#22203)
5 years ago
Tao Luo 01fa4ead61
fix -Wno-error=sign-compare warning in gcc8 (#21434)
5 years ago
zhaoyuchen2018 b16274556a
Add dscending for argsort (#21400)
5 years ago
Yibing Liu 98460c009e Simplify the computation in cpu
7 years ago
Yibing Liu 94e72ea6e7 Support more negative axes in argsort_op
7 years ago
Yibing Liu 2c2120c8a2 Remove redundant code
7 years ago
Yibing Liu 4760f2851e Add the argsort operator
7 years ago