tf.compat.v1.arg_max