Commit Graph

2 Commits (b2f7ab6636d5e7fcc3bfd655c071416f190ed619)

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