diff options
Diffstat (limited to 'tensorflow/python/data/experimental/kernel_tests/optimization/noop_elimination_test.py')
| -rw-r--r-- | tensorflow/python/data/experimental/kernel_tests/optimization/noop_elimination_test.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/data/experimental/kernel_tests/optimization/noop_elimination_test.py b/tensorflow/python/data/experimental/kernel_tests/optimization/noop_elimination_test.py index 092e0ff62a..fb0640fe9f 100644 --- a/tensorflow/python/data/experimental/kernel_tests/optimization/noop_elimination_test.py +++ b/tensorflow/python/data/experimental/kernel_tests/optimization/noop_elimination_test.py @@ -40,7 +40,9 @@ class NoopEliminationTest(test_base.DatasetTestBase): ["FiniteRepeat", "FiniteSkip", "Prefetch", "Prefetch"])) dataset = dataset.repeat(some_tensor).skip(5).prefetch(0).take(-1).skip( 0).repeat(1).prefetch(0) - dataset = dataset.apply(optimization.optimize(["noop_elimination"])) + options = dataset_ops.Options() + options.experimental_noop_elimination = True + dataset = dataset.with_options(options) iterator = dataset.make_one_shot_iterator() get_next = iterator.get_next() |