message: Enhancement with deprecated_argument_lookup for argmax
  This fix makes some enhancement for argmax, using
  deprecated_argument_lookup instread of customerized logic.
diff: @@ -208,11 +208,9 @@ def argmax(input,
  name=None,
  dimension=None,
  output_type=dtypes.int64):
  - if dimension is not None:
  - if axis is not None:
  - raise ValueError("Cannot specify both 'axis' and 'dimension'")
  - axis = dimension
  - elif axis is None:
  + axis = deprecation.deprecated_argument_lookup(
  + "axis", axis, "dimension", dimension)
  + if axis is None:
  axis = 0
  return gen_math_ops.arg_max(input, axis, name=name, output_type=output_type)
  