%cst = arith.constant dense<30> : tensor<i64>
    %cst_1 = arith.constant dense<-1> : tensor<i64>
    %cst_2 = arith.constant dense<165000> : tensor<i64>
    %0 = "tf.TakeDataset"(%arg0, %cst) {device = "", metadata = "", output_shapes = [#tf_type.shape<>], output_types = [!tf_type.string]} : (tensor<!tf_type.variant>, tensor<i64>) -> tensor<!tf_type.variant>

  }];

  let arguments = (ins);

  let results = (outs
    TF_Int32Tensor:$device_ordinal
  );
}

def TF_TakeDatasetOp : TF_Op<"TakeDataset", [Pure]> {
  let summary = [{
Creates a dataset that contains `count` elements from the `input_dataset`.
  }];

  let arguments = (ins
    TF_VariantTensor:$input_dataset,