compute-argmax