Commit Graph

5 Commits (e92f770c425d7ebf0056b98a59756a321d8ddb88)

Author SHA1 Message Date
Zhaolong Xing f00f982a02
add cub impl for arg max, min (#25941)
5 years ago
Yu Yang 9bd70a1e04 Change tensor uses proto::VarType::type
7 years ago
yuyang18 9b43edeae0 Polish arg_min_max_op
7 years ago
sneaxiy 6d32e96096 remove redundant comments
7 years ago
sneaxiy 568c4e5ec4 recommit using account sneaxiy
7 years ago