message: undo not required deletion
diff: @@ -968,7 +968,8 @@ REGISTER_OP("ArgMax")
  .Output("output: output_type")
  .Attr("T: numbertype")
  .Attr("Tidx: {int32, int64} = DT_INT32")
  - .Attr("output_type: {int32, int64} = DT_INT64");
  + .Attr("output_type: {int32, int64} = DT_INT64")
  + .SetShapeFn(ArgOpShape);
  REGISTER_OP("ArgMin")
  @@ -977,7 +978,8 @@ REGISTER_OP("ArgMin")
  .Output("output: output_type")
  .Attr("T: numbertype")
  .Attr("Tidx: {int32, int64} = DT_INT32")
  - .Attr("output_type: {int32, int64} = DT_INT64");
  + .Attr("output_type: {int32, int64} = DT_INT64")
  + .SetShapeFn(ArgOpShape);
  namespace {
  