aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/autograph
diff options
context:
space:
mode:
authorGravatar feiquan <feiquan@wacai.com>2018-08-15 08:47:22 +0800
committerGravatar feiquan <feiquan@wacai.com>2018-08-15 08:47:22 +0800
commitf2134cbd2ec4dd98f9f20ac41e4f46cdd0246af2 (patch)
tree83989a84b71d7a08aa9801a573e37c4183f9f64d /tensorflow/contrib/autograph
parent349d81c80a5b64ae09a36624571ec24d9e7a8b1d (diff)
use get_item_tensor_string for string with rank 0
Diffstat (limited to 'tensorflow/contrib/autograph')
-rw-r--r--tensorflow/contrib/autograph/operators/slices_test.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/contrib/autograph/operators/slices_test.py b/tensorflow/contrib/autograph/operators/slices_test.py
index 9c0b2c77a1..5300428462 100644
--- a/tensorflow/contrib/autograph/operators/slices_test.py
+++ b/tensorflow/contrib/autograph/operators/slices_test.py
@@ -53,6 +53,12 @@ class SlicesTest(test.TestCase):
with self.test_session() as sess:
self.assertEqual(sess.run(t), b"b")
+ initial_list_str = constant_op.constant(["abcd", "bcde"])
+ t = slices.get_item(initial_list_str, 1, slices.GetItemOpts(element_dtype=initial_str.dtype))
+
+ with self.test_session() as sess:
+ self.assertEqual(sess.run(t), b"bcde")
+
if __name__ == '__main__':
test.main()