meridian.backend.argmax

TensorFlow implementation for argmax.