updated the condition if dtype is int32

tensorflow · Dec 24, 2022 · 3fb2931 · 3fb2931
1 parent 17f1e40
commit 3fb2931
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/python/framework/ops.py b/tensorflow/python/framework/ops.py
@@ -1597,10 +1597,10 @@ def convert_to_tensor(value,
   if isinstance(value, Tensor):
     if dtype is not None:
       #checking for integer overflow
-      if (abs(value) > (1 << 31) - 1):
+      if (dtype == dtypes.int32) and (abs(value) > (1 << 31) - 1):
         raise RuntimeError(
                 _add_error_prefix(
-                    "integer overflow for dtype int32",
+                    f"input value exceeds int32 dtype limit",
                     name=name))
       if not dtype.is_compatible_with(value.dtype):
         raise ValueError(