def TF_RepeatDatasetOp : TF_Op<"RepeatDataset", [Pure]> {
  let summary = [{
Creates a dataset that emits the outputs of `input_dataset` `count` times.
  }];

  let arguments = (ins
    TF_VariantTensor:$input_dataset,
    Arg<TF_Int64Tensor, [{A scalar representing the number of times that `input_dataset` should
be repeated. A value of `-1` indicates that it should be repeated infinitely.}]>:$count,