message: Added reference type support for remaining TF ops
  These ops were not created with the latest TF dialect generation script which
  automatically adds reference type support now.
diff: @@ -16,12 +16,12 @@ limitations under the License.
  // This is the operation definition file for TensorFlow.
  //
  // This file contains TensorFlow ops whose definitions are programmatically
  -// generated from the TensorFlow codebase from the ApiDef protos under
  -// tensorflow/core/api_def/base_api. The generated fields for an op
  -// includes name, summary, description, traits, arguments, results, derived
  -// attributes. Therefore, modifications to these fields will **not** be
  -// respected upon subsequent refreshes. However, additional fields after those
  -// fields will be retained.
  +// generated from the api-def-files in the following folder:
  +// tensorflow/core/api_def/base_api
  +// The generated fields for an op include name, summary, description, traits,
  +// arguments, results, derived attributes. Therefore, modifications to these
  +// fields will **not** be respected upon subsequent refreshes. However,
  +// additional fields after those fields will be retained.
  //
  // If you absolutely need to modify the generated fields of an op, move the
  // definition to `tf_ops.td` and perform the modification there.
  @@ -311,8 +311,8 @@ def TF_AnonymousRandomSeedGeneratorOp : TF_Op<"AnonymousRandomSeedGenerator", []
  let summary = "";
  let arguments = (ins
  - I64Tensor:$seed,
  - I64Tensor:$seed2
  + TF_Int64Tensor:$seed,
  + TF_Int64Tensor:$seed2
  );
  let results = (outs
  @@ -325,9 +325,9 @@ def TF_AnonymousSeedGeneratorOp : TF_Op<"AnonymousSeedGenerator", []> {
  let summary = "";
  let arguments = (ins
  - I64Tensor:$seed,
  - I64Tensor:$seed2,
  - I1Tensor:$reshuffle
  + TF_Int64Tensor:$seed,
  + TF_Int64Tensor:$seed2,
  + TF_BoolTensor:$reshuffle
  );
  let results = (outs
  @@ -10276,10 +10276,10 @@ def TF_ShuffleAndRepeatDatasetV2Op : TF_Op<"ShuffleAndRepeatDatasetV2", []> {
  let arguments = (ins
  TF_VariantTensor:$input_dataset,
  - I64Tensor:$buffer_size,
  - I64Tensor:$seed,
  - I64Tensor:$seed2,
  - I64Tensor:$count,
  + TF_Int64Tensor:$buffer_size,
  + TF_Int64Tensor:$seed,
  + TF_Int64Tensor:$seed2,
  + TF_Int64Tensor:$count,
  Arg<TF_ResourceTensor, "", [TF_DatasetSeedGeneratorRead, TF_DatasetSeedGeneratorWrite]>:$seed_generator,
  DefaultValuedAttr<BoolAttr, "true">:$reshuffle_each_iteration,
  @@ -10297,7 +10297,7 @@ def TF_ShuffleDatasetV2Op : TF_Op<"ShuffleDatasetV2", []> {
  let arguments = (ins
  TF_VariantTensor:$input_dataset,
  - I64Tensor:$buffer_size,
  + TF_Int64Tensor:$buffer_size,
  Arg<TF_ResourceTensor, "", [TF_DatasetSeedGeneratorRead, TF_DatasetSeedGeneratorWrite]>:$seed_generator,
  Confined<TypeArrayAttr, [ArrayMinCount<1>]>:$output_types,
  @@ -10314,9 +10314,9 @@ def TF_ShuffleDatasetV3Op : TF_Op<"ShuffleDatasetV3", []> {
  let arguments = (ins
  TF_VariantTensor:$input_dataset,
  - I64Tensor:$buffer_size,
  - I64Tensor:$seed,
  - I64Tensor:$seed2,
  + TF_Int64Tensor:$buffer_size,
  + TF_Int64Tensor:$seed,
  + TF_Int64Tensor:$seed2,
  Arg<TF_ResourceTensor, "", [TF_DatasetSeedGeneratorRead, TF_DatasetSeedGeneratorWrite]>:$seed_generator,
  DefaultValuedAttr<BoolAttr, "true">:$reshuffle_each_iteration,
  