Skip to content

Performance of argmax in different systems/libraries

Notifications You must be signed in to change notification settings

intoraw/argmax-perf

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Compare argmax op in different system/libries

  • Eigen : tensorflow's choice, with tensor support
  • mxnet : handcrafted argmax, with tensor support
  • cublas : amax, absolute max, not argmax, only support vector
  • cub : nvidia-cub, support vector and batch-vector

cub has the best performance than others

About

Performance of argmax in different systems/libraries

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published