From 435f4605b7d1e7d2c9d7937ab9c0cbf4069bdc74 Mon Sep 17 00:00:00 2001 From: Derek Murray Date: Mon, 19 Nov 2018 19:40:01 -0800 Subject: [PATCH] [tf.data] Shard the map_vectorization_test to avoid timeouts. PiperOrigin-RevId: 222179740 --- .../python/data/experimental/kernel_tests/optimization/BUILD | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/python/data/experimental/kernel_tests/optimization/BUILD b/tensorflow/python/data/experimental/kernel_tests/optimization/BUILD index 9946ef5a42f..1d0e6af6493 100644 --- a/tensorflow/python/data/experimental/kernel_tests/optimization/BUILD +++ b/tensorflow/python/data/experimental/kernel_tests/optimization/BUILD @@ -202,6 +202,7 @@ py_test( name = "map_vectorization_test", size = "medium", srcs = ["map_vectorization_test.py"], + shard_count = 8, srcs_version = "PY2AND3", tags = [ "no_oss",