Commit Graph

2 Commits (11adb0f373d20c598701fafa41f715d4db37672f)

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