@tf_export(v1=['math.argmax', 'argmax'])
@deprecation.deprecated_args(None, 'Use the `axis` argument instead', 'dimension')
@_set_doc(gen_math_ops.arg_max.__doc__.replace('dimensions', 'axes').replace('dimension', 'axis'))
def argmax(input, axis=None, name=None, dimension=None, output_type=dtypes.int64):
    axis = deprecation.deprecated_argument_lookup('axis', axis, 'dimension', dimension)
    return argmax_v2(input, axis, output_type, name)
@tf_export(v1=['math.argmax', 'argmax'])
@deprecation.deprecated_args(None, 'Use the `axis` argument instead', 'dimension')
@_set_doc(gen_math_ops.arg_max.__doc__.replace('dimensions', 'axes').replace('dimension', 'axis'))
def argmax(input, axis=None, name=None, dimension=None, output_type=dtypes.int64):
    axis = deprecation.deprecated_argument_lookup('axis', axis, 'dimension', dimension)
    return argmax_v2(input, axis, output_type, name)