Commit Graph

2 Commits (4f9d6529feb259c148ab2b2bf80643a211fb62ae)

Author SHA1 Message Date
wawltor 6b28456ed0
add the argmax, argmin for the api2.0
5 years ago
Zhaolong Xing f00f982a02
add cub impl for arg max, min (#25941)
5 years ago