diff options
author | 2018-09-04 14:01:13 -0700 | |
---|---|---|
committer | 2018-09-04 14:17:53 -0700 | |
commit | 06e8109af2e5ae5bc149e25fc64fbf66d6c8b817 (patch) | |
tree | b13b214063b3a4ba8f15c26e9170c9f44c49b854 /tensorflow/core/ops/dataset_ops.cc | |
parent | 8ef276fd2181fb71c2e232f60aa45ee96cb5905b (diff) |
[tf.data] Add internal optimizations for executing simple functions in `MapDataset`.
PiperOrigin-RevId: 211520001
Diffstat (limited to 'tensorflow/core/ops/dataset_ops.cc')
-rw-r--r-- | tensorflow/core/ops/dataset_ops.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/ops/dataset_ops.cc b/tensorflow/core/ops/dataset_ops.cc index f03639e833..1a5ad8f421 100644 --- a/tensorflow/core/ops/dataset_ops.cc +++ b/tensorflow/core/ops/dataset_ops.cc @@ -198,6 +198,7 @@ REGISTER_OP("MapDataset") .Attr("Targuments: list(type) >= 0") .Attr("output_types: list(type) >= 1") .Attr("output_shapes: list(shape) >= 1") + .Attr("use_inter_op_parallelism: bool = true") .SetShapeFn(shape_inference::ScalarShape); REGISTER_OP("ParallelMapDataset") |