Auto generate TensorFlow WriteSummary ops.

Add summary and descriptions for these ops in api definition.

PiperOrigin-RevId: 326467435
Change-Id: I8037543a925fd7ba254041aa270f45afc4553dc6

tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

@@ -11517,6 +11517,140 @@ where(input) ==> [[0, 0, 0],
   TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
 }
 
+def TF_WriteAudioSummaryOp : TF_Op<"WriteAudioSummary", []> {
+  let summary = "Writes an audio summary.";
+
+  let description = [{
+Writes encoded audio summary `tensor` at `step` with `tag` using summary `writer`.
+`sample_rate` is the audio sample rate is Hz.
+  }];
+
+  let arguments = (ins
+    TF_ResourceTensor:$writer,
+    I64Tensor:$step,
+    TF_StrTensor:$tag,
+    F32Tensor:$tensor,
+    F32Tensor:$sample_rate,
+
+    Confined<DefaultValuedAttr<I64Attr, "3">, [IntMinValue<1>]>:$max_outputs
+  );
+
+  let results = (outs);
+}
+
+def TF_WriteGraphSummaryOp : TF_Op<"WriteGraphSummary", []> {
+  let summary = "Writes a graph summary.";
+
+  let description = [{
+Writes TensorFlow graph `tensor` at `step` using summary `writer`.
+  }];
+
+  let arguments = (ins
+    TF_ResourceTensor:$writer,
+    I64Tensor:$step,
+    TF_StrTensor:$tensor
+  );
+
+  let results = (outs);
+}
+
+def TF_WriteHistogramSummaryOp : TF_Op<"WriteHistogramSummary", []> {
+  let summary = "Writes a histogram summary.";
+
+  let description = [{
+Writes histogram `values` at `step` with `tag` using summary `writer`.
+  }];
+
+  let arguments = (ins
+    TF_ResourceTensor:$writer,
+    I64Tensor:$step,
+    TF_StrTensor:$tag,
+    TF_IntOrFpTensor:$values
+  );
+
+  let results = (outs);
+
+  TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<3>;
+}
+
+def TF_WriteImageSummaryOp : TF_Op<"WriteImageSummary", []> {
+  let summary = "Writes an image summary.";
+
+  let description = [{
+Writes image `tensor` at `step` with `tag` using summary `writer`.
+`tensor` is image with shape [height, width, channels].
+  }];
+
+  let arguments = (ins
+    TF_ResourceTensor:$writer,
+    I64Tensor:$step,
+    TF_StrTensor:$tag,
+    TensorOf<[F16, F32, TF_Uint8]>:$tensor,
+    TF_Uint8Tensor:$bad_color,
+
+    Confined<DefaultValuedAttr<I64Attr, "3">, [IntMinValue<1>]>:$max_images
+  );
+
+  let results = (outs);
+
+  TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<3>;
+}
+
+def TF_WriteRawProtoSummaryOp : TF_Op<"WriteRawProtoSummary", []> {
+  let summary = "Writes a serialized proto summary.";
+
+  let description = [{
+Writes `tensor`, a serialized proto at `step` using summary `writer`.
+  }];
+
+  let arguments = (ins
+    TF_ResourceTensor:$writer,
+    I64Tensor:$step,
+    TF_StrTensor:$tensor
+  );
+
+  let results = (outs);
+}
+
+def TF_WriteScalarSummaryOp : TF_Op<"WriteScalarSummary", []> {
+  let summary = "Writes a scalar summary.";
+
+  let description = [{
+Writes scalar `value` at `step` with `tag` using summary `writer`.
+  }];
+
+  let arguments = (ins
+    TF_ResourceTensor:$writer,
+    I64Tensor:$step,
+    TF_StrTensor:$tag,
+    TF_IntOrFpTensor:$value
+  );
+
+  let results = (outs);
+
+  TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<3>;
+}
+
+def TF_WriteSummaryOp : TF_Op<"WriteSummary", []> {
+  let summary = "Writes a tensor summary.";
+
+  let description = [{
+Writes `tensor` at `step` with `tag` using summary `writer`.
+  }];
+
+  let arguments = (ins
+    TF_ResourceTensor:$writer,
+    I64Tensor:$step,
+    TF_Tensor:$tensor,
+    TF_StrTensor:$tag,
+    TF_StrTensor:$summary_metadata
+  );
+
+  let results = (outs);
+
+  TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<2>;
+}
+
 def TF_XdivyOp : TF_Op<"Xdivy", [NoSideEffect, ResultsBroadcastableShape, TF_SameOperandsAndResultElementTypeResolveRef]>,
                  WithBroadcastableBinOpBuilder {
   let summary = "Returns 0 if x == 0, and x / y otherwise, elementwise.";

tensorflow/core/api_def/base_api/api_def_WriteAudioSummary.pbtxt

@@ -1,4 +1,9 @@
 op {
   graph_op_name: "WriteAudioSummary"
   visibility: HIDDEN
+  summary: "Writes an audio summary."
+  description: <<END
+Writes encoded audio summary `tensor` at `step` with `tag` using summary `writer`.
+`sample_rate` is the audio sample rate is Hz.
+END
 }

tensorflow/core/api_def/base_api/api_def_WriteGraphSummary.pbtxt

@@ -1,4 +1,8 @@
 op {
   graph_op_name: "WriteGraphSummary"
   visibility: HIDDEN
+  summary: "Writes a graph summary."
+  description: <<END
+Writes TensorFlow graph `tensor` at `step` using summary `writer`.
+END
 }

tensorflow/core/api_def/base_api/api_def_WriteHistogramSummary.pbtxt

@@ -1,4 +1,8 @@
 op {
   graph_op_name: "WriteHistogramSummary"
   visibility: HIDDEN
+  summary: "Writes a histogram summary."
+  description: <<END
+Writes histogram `values` at `step` with `tag` using summary `writer`.
+END
 }

tensorflow/core/api_def/base_api/api_def_WriteImageSummary.pbtxt

@@ -1,4 +1,9 @@
 op {
   graph_op_name: "WriteImageSummary"
   visibility: HIDDEN
+  summary: "Writes an image summary."
+  description: <<END
+Writes image `tensor` at `step` with `tag` using summary `writer`.
+`tensor` is image with shape [height, width, channels].
+END
 }

tensorflow/core/api_def/base_api/api_def_WriteRawProtoSummary.pbtxt

@@ -1,4 +1,8 @@
 op {
   graph_op_name: "WriteRawProtoSummary"
   visibility: HIDDEN
+  summary: "Writes a serialized proto summary."
+  description: <<END
+Writes `tensor`, a serialized proto at `step` using summary `writer`.
+END
 }

tensorflow/core/api_def/base_api/api_def_WriteScalarSummary.pbtxt

@@ -1,4 +1,8 @@
 op {
   graph_op_name: "WriteScalarSummary"
   visibility: HIDDEN
+  summary: "Writes a scalar summary."
+  description: <<END
+Writes scalar `value` at `step` with `tag` using summary `writer`.
+END
 }

tensorflow/core/api_def/base_api/api_def_WriteSummary.pbtxt

@@ -1,4 +1,8 @@
 op {
   graph_op_name: "WriteSummary"
   visibility: HIDDEN
+  summary: "Writes a tensor summary."
+  description: <<END
+Writes `tensor` at `step` with `tag` using summary `writer`.
+END
 }