From dc3f225de9b5d969ccb60871926f37dd7aaf7863 Mon Sep 17 00:00:00 2001 From: Ken Franko Date: Thu, 13 Aug 2020 09:56:40 -0700 Subject: [PATCH] Auto generate TensorFlow WriteSummary ops. Add summary and descriptions for these ops in api definition. PiperOrigin-RevId: 326467435 Change-Id: I8037543a925fd7ba254041aa270f45afc4553dc6 --- .../mlir/tensorflow/ir/tf_generated_ops.td | 134 ++++++++++++++++++ .../base_api/api_def_WriteAudioSummary.pbtxt | 5 + .../base_api/api_def_WriteGraphSummary.pbtxt | 4 + .../api_def_WriteHistogramSummary.pbtxt | 4 + .../base_api/api_def_WriteImageSummary.pbtxt | 5 + .../api_def_WriteRawProtoSummary.pbtxt | 4 + .../base_api/api_def_WriteScalarSummary.pbtxt | 4 + .../base_api/api_def_WriteSummary.pbtxt | 4 + 8 files changed, 164 insertions(+) diff --git a/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td b/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td index cd8df635cba..cc07d50eee2 100644 --- a/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td +++ b/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, [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, [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."; diff --git a/tensorflow/core/api_def/base_api/api_def_WriteAudioSummary.pbtxt b/tensorflow/core/api_def/base_api/api_def_WriteAudioSummary.pbtxt index 520952cd411..e7e0a95cabc 100644 --- a/tensorflow/core/api_def/base_api/api_def_WriteAudioSummary.pbtxt +++ b/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: <