diff options
| author |  Rachel Lim <rachelim@google.com> | 2018-10-09 14:36:33 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 14:46:11 -0700 |
| commit | 950cf87104bfee28e2165fe368f66337b8a1336d (patch) | |
| tree | 59ace2e229776b79897c54b4be0705231d5ac9f2 /tensorflow/python/data/experimental/kernel_tests/optimization/map_vectorization_test.py | |
| parent | 35caff957424a60bd7d7e4e92a1ec87f617781c6 (diff) | |
[tf.data vectorization] Add vectorizer for `Add` op
PiperOrigin-RevId: 216424512
Diffstat (limited to 'tensorflow/python/data/experimental/kernel_tests/optimization/map_vectorization_test.py')
| -rw-r--r-- | tensorflow/python/data/experimental/kernel_tests/optimization/map_vectorization_test.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/data/experimental/kernel_tests/optimization/map_vectorization_test.py b/tensorflow/python/data/experimental/kernel_tests/optimization/map_vectorization_test.py index 803ff87924..d1d6cf28ab 100644 --- a/tensorflow/python/data/experimental/kernel_tests/optimization/map_vectorization_test.py +++ b/tensorflow/python/data/experimental/kernel_tests/optimization/map_vectorization_test.py @@ -80,6 +80,7 @@ class MapVectorizationTest(test_base.DatasetTestBase, parameterized.TestCase): ("Basic", lambda x: (x, x + 1), None), ("Const", lambda x: 2, 12), ("Parallel", lambda x: (x, x + 1), 12), + ("Broadcast", lambda x: x + np.random.rand(5, 4, 3, 2), None), ("Gather", lambda x: array_ops.gather(x, 0), 12), ) def testOptimization(self, map_fn, num_parallel_calls): |