From 2d67d32e587e773811da4577587dc07cf2922641 Mon Sep 17 00:00:00 2001
From: Yong Tang <yong.tang.github@outlook.com>
Date: Thu, 8 Mar 2018 21:57:07 +0000
Subject: [PATCH] Update api defs

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
---
 .../api_def_DrawBoundingBoxesV2.pbtxt         | 43 +++++++++++++++++++
 tensorflow/core/ops/image_ops.cc              | 24 +----------
 2 files changed, 44 insertions(+), 23 deletions(-)
 create mode 100644 tensorflow/core/api_def/base_api/api_def_DrawBoundingBoxesV2.pbtxt

diff --git a/tensorflow/core/api_def/base_api/api_def_DrawBoundingBoxesV2.pbtxt b/tensorflow/core/api_def/base_api/api_def_DrawBoundingBoxesV2.pbtxt
new file mode 100644
index 00000000000..5c3edb5cb32
--- /dev/null
+++ b/tensorflow/core/api_def/base_api/api_def_DrawBoundingBoxesV2.pbtxt
@@ -0,0 +1,43 @@
+op {
+  graph_op_name: "DrawBoundingBoxesV2"
+  in_arg {
+    name: "images"
+    description: <<END
+4-D with shape `[batch, height, width, depth]`. A batch of images.
+END
+  }
+  in_arg {
+    name: "boxes"
+    description: <<END
+3-D with shape `[batch, num_bounding_boxes, 4]` containing bounding
+boxes.
+END
+  }
+  in_arg {
+    name: "colors"
+    description: <<END
+2-D. A list of RGBA colors to cycle through for the boxes.
+END
+  }
+  out_arg {
+    name: "output"
+    description: <<END
+4-D with the same shape as `images`. The batch of input images with
+bounding boxes drawn on the images.
+END
+  }
+  summary: "Draw bounding boxes on a batch of images."
+  description: <<END
+Outputs a copy of `images` but draws on top of the pixels zero or more bounding
+boxes specified by the locations in `boxes`. The coordinates of the each
+bounding box in `boxes` are encoded as `[y_min, x_min, y_max, x_max]`. The
+bounding box coordinates are floats in `[0.0, 1.0]` relative to the width and
+height of the underlying image.
+
+For example, if an image is 100 x 200 pixels (height x width) and the bounding
+box is `[0.1, 0.2, 0.5, 0.9]`, the upper-left and bottom-right coordinates of
+the bounding box will be `(40, 10)` to `(100, 50)` (in (x,y) coordinates).
+
+Parts of the bounding box may fall outside the image.
+END
+}
diff --git a/tensorflow/core/ops/image_ops.cc b/tensorflow/core/ops/image_ops.cc
index 07dd4ab1602..b8769355308 100644
--- a/tensorflow/core/ops/image_ops.cc
+++ b/tensorflow/core/ops/image_ops.cc
@@ -603,29 +603,7 @@ REGISTER_OP("DrawBoundingBoxesV2")
     .Attr("T: {float, half} = DT_FLOAT")
     .SetShapeFn([](InferenceContext* c) {
       return shape_inference::UnchangedShapeWithRankAtLeast(c, 3);
-    })
-    .Doc(R"doc(
-Draw bounding boxes on a batch of images.
-
-Outputs a copy of `images` but draws on top of the pixels zero or more bounding
-boxes specified by the locations in `boxes`. The coordinates of the each
-bounding box in `boxes` are encoded as `[y_min, x_min, y_max, x_max]`. The
-bounding box coordinates are floats in `[0.0, 1.0]` relative to the width and
-height of the underlying image.
-
-For example, if an image is 100 x 200 pixels (height x width) and the bounding
-box is `[0.1, 0.2, 0.5, 0.9]`, the upper-left and bottom-right coordinates of
-the bounding box will be `(40, 10)` to `(100, 50)` (in (x,y) coordinates).
-
-Parts of the bounding box may fall outside the image.
-
-images: 4-D with shape `[batch, height, width, depth]`. A batch of images.
-boxes: 3-D with shape `[batch, num_bounding_boxes, 4]` containing bounding
-  boxes.
-colors: 2-D. A list of RGBA colors to cycle through for the boxes.
-output: 4-D with the same shape as `images`. The batch of input images with
-  bounding boxes drawn on the images.
-)doc");
+    });
 
 // --------------------------------------------------------------------------
 REGISTER_OP("SampleDistortedBoundingBox")