Commit Graph

2 Commits (ebf9b2125ec61e4e016fa79890a194b295cdde95)

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