Commit Graph

2 Commits (c0163837a5267e989d1fd040e2ce9248f2fa68b3)

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