add tensorflow.math.round

Author: hoel-bagard

stubs/tensorflow/tensorflow/__init__.pyi

@@ -56,6 +56,7 @@ from tensorflow.math import (
     reduce_min as reduce_min,
     reduce_prod as reduce_prod,
     reduce_sum as reduce_sum,
+    round as round,
     sigmoid as sigmoid,
     sign as sign,
     sin as sin,

stubs/tensorflow/tensorflow/math.pyi

@@ -219,6 +219,12 @@ def square(x: RaggedTensor, name: str | None = None) -> RaggedTensor: ...
 def softplus(features: TensorCompatible, name: str | None = None) -> Tensor: ...
 @overload
 def softplus(features: RaggedTensor, name: str | None = None) -> RaggedTensor: ...
+@overload
+def round(x: TensorCompatible, name: str | None = None) -> Tensor: ...
+@overload
+def round(x: SparseTensor, name: str | None = None) -> SparseTensor: ...
+@overload
+def round(x: RaggedTensor, name: str | None = None) -> RaggedTensor: ...
 
 # Depending on the method axis is either a rank 0 tensor or a rank 0/1 tensor.
 def reduce_mean(