diff options
author | Piotr Padlewski <prazek@google.com> | 2018-09-14 11:28:28 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-14 11:32:52 -0700 |
commit | c20a7b81d79d30db9e990309ddb419bcb48120cc (patch) | |
tree | 9ea682cf79bac18653e7690785e0f5e7117b6b8b /tensorflow/core/BUILD | |
parent | 89f9080ed0d1a43cb2fa253997b2553c6916f364 (diff) |
[tf.data] Introducing an optimization that parallelizes map transformations.
Stateless MapDatasets can be paralellized by switching to ParallelMapDataset. We set `num_parallel_calls` to 2 for now, but in the future a special value will be used that result in the optimal value to be selected dynamically at runtime.
This patch also exposed a memory leak which was fixed.
PiperOrigin-RevId: 213015223
Diffstat (limited to 'tensorflow/core/BUILD')
-rw-r--r-- | tensorflow/core/BUILD | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 1a86bff5cd..55715bb3a6 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -1429,9 +1429,11 @@ cc_library( ":test", ":testlib_ops", "//tensorflow/cc:scope", + "//tensorflow/core/kernels:cast_op", "//tensorflow/core/kernels:constant_op", "//tensorflow/core/kernels:ops_testutil", "//tensorflow/core/kernels:ops_util", + "//tensorflow/core/kernels:random_ops", ], ) |