aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/tests/tensor_array_ops_test.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-12-14 11:18:05 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-12-14 11:21:42 -0800
commit9bbd25da077881696875447c3081f96c20e8728c (patch)
treebc55d97845e0acd017d90cc077bdfd8e89d4d049 /tensorflow/compiler/tests/tensor_array_ops_test.py
parent9277bb73a926684d4346a56fec6c117873a9a84a (diff)
Enable bfloat16 tests and add a filter for currently
failed tests. PiperOrigin-RevId: 179069257
Diffstat (limited to 'tensorflow/compiler/tests/tensor_array_ops_test.py')
-rw-r--r--tensorflow/compiler/tests/tensor_array_ops_test.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/compiler/tests/tensor_array_ops_test.py b/tensorflow/compiler/tests/tensor_array_ops_test.py
index ac039e0162..a62925a181 100644
--- a/tensorflow/compiler/tests/tensor_array_ops_test.py
+++ b/tensorflow/compiler/tests/tensor_array_ops_test.py
@@ -330,8 +330,7 @@ class TensorArrayTest(xla_test.XLATestCase):
# Find two different floating point types, create an array of
# the first type, but try to read the other type.
if len(self.float_types) > 1:
- dtype1 = self.float_types[0]
- dtype2 = self.float_types[1]
+ dtype1, dtype2 = list(self.float_types)[:2]
with self.test_session(), self.test_scope():
ta = tensor_array_ops.TensorArray(
dtype=dtype1, tensor_array_name="foo", size=3)