Added support for argmax/argmin
9 files changed