experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

The tf.data is working on changes which will make it no longer possible to set tf.data.Options by mutating tf.data.Dataset.options(). In preparation for this change, we plan to make the value returned by tf.data.Dataset.options() immutable. We have identified that the code in this CL would be affected and fix it beforehand.

Browse Source 
PiperOrigin-RevId: 361202813
Change-Id: Ib383fa360d9e80cddd3cdcae784bd0db75393d8c
This commit is contained in:
A. Unique TensorFlower 2021-03-05 12:35:45 -08:00 committed by TensorFlower Gardener
parent e9dac2c6f8
commit 435c155dbe
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/data/benchmarks/benchmark_base.py
View File

@ -171,7 +171,7 @@ class DatasetBenchmarkBase(test.Benchmark):
# The options that have been applied to the dataset are preserved so that
# they are not overwritten while benchmarking.
options = dataset.options()
options = dataset_ops.Options()
options.experimental_optimization.apply_default_optimizations = (
apply_default_optimizations)
dataset = dataset.with_options(options)