Update testTensorArraySplitRead to not overflow for int8.

kenfranko · tensorflower-gardener · commit 234a51acec19 · 2021-08-03T15:22:40.000-07:00
Overflow behavior for unsigned int in C.

PiperOrigin-RevId: 388555897
Change-Id: Ie65e9256fbd9bf52f0c7c4f1c768b94db738fdc1
diff --git a/tensorflow/compiler/tests/tensor_array_ops_test.py b/tensorflow/compiler/tests/tensor_array_ops_test.py
@@ -247,7 +247,7 @@ def fn():
             dtype=tf_dtype, tensor_array_name="foo", size=3)
         lengths = constant_op.constant([1, 1, 1])
         w0 = ta.split(
-            convert([[1.0, 101.0], [2.0, 201.0], [3.0, 301.0]]),
+            convert([[1.0, 101.0], [2.0, 121.0], [3.0, 127.0]]),
             lengths=lengths)
         r0 = w0.read(0)
         r1 = w0.read(1)
@@ -256,14 +256,13 @@ def fn():
 
       d0, d1, d2 = self.evaluate(xla.compile(fn))
       self.assertAllEqual(convert([[1.0, 101.0]]), d0)
-      self.assertAllEqual(convert([[2.0, 201.0]]), d1)
-      self.assertAllEqual(convert([[3.0, 301.0]]), d2)
+      self.assertAllEqual(convert([[2.0, 121.0]]), d1)
+      self.assertAllEqual(convert([[3.0, 127.0]]), d2)
 
-  # Disable temporarily due to b/195023333
-  # @test_util.disable_control_flow_v2("b/122315872 (split)")
-  # def testTensorArraySplitRead(self):
-  #   for dtype in self.numeric_tf_types:
-  #     self._testTensorArraySplitRead(dtype)
+  @test_util.disable_control_flow_v2("b/122315872 (split)")
+  def testTensorArraySplitRead(self):
+    for dtype in self.numeric_tf_types:
+      self._testTensorArraySplitRead(dtype)
 
   @test_util.disable_control_flow_v2("TensorArray.grad is not supported in v2")
   def testTensorGradArrayWriteRead(self):
