CUDA argmax layer.