Update docstring for tf.round as it support all real types.
authorYong Tang <yong.tang.github@outlook.com>
Mon, 21 May 2018 23:57:17 +0000 (23:57 +0000)
committerYong Tang <yong.tang.github@outlook.com>
Mon, 21 May 2018 23:57:17 +0000 (23:57 +0000)
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
tensorflow/python/ops/math_ops.py

index 80351a8..776f9b8 100644 (file)
@@ -600,7 +600,7 @@ def round(x, name=None):  # pylint: disable=redefined-builtin
   ```
 
   Args:
-    x: A `Tensor` of type `float32` or `float64`.
+    x: A `Tensor` of type `float16`, `float32`, `float64`, `int32`, or `int64`.
     name: A name for the operation (optional).
 
   Returns: