diff options
author | 2018-09-26 09:47:58 -0700 | |
---|---|---|
committer | 2018-09-26 09:51:57 -0700 | |
commit | 23a07f2c1444509986eece54e486cdcf0b8e32e4 (patch) | |
tree | 1ecd077dc46e06295d9f0a5cea720d7e56ddd1df /tensorflow/contrib/data | |
parent | eac28534e883283977ebae4dc4dea00cdd601fbc (diff) |
[tf.data] Adding serialization support for `StatsAggregatorDatasets` to make it possible to apply static optimizations to input pipelines whose prefix contains the `set_stats_aggregator` transformation.
PiperOrigin-RevId: 214619583
Diffstat (limited to 'tensorflow/contrib/data')
-rw-r--r-- | tensorflow/contrib/data/python/kernel_tests/optimization/latency_all_edges_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/data/python/kernel_tests/optimization/latency_all_edges_test.py b/tensorflow/contrib/data/python/kernel_tests/optimization/latency_all_edges_test.py index db380c02a9..e4f18222fd 100644 --- a/tensorflow/contrib/data/python/kernel_tests/optimization/latency_all_edges_test.py +++ b/tensorflow/contrib/data/python/kernel_tests/optimization/latency_all_edges_test.py @@ -34,8 +34,8 @@ class OptimizeStatsDatasetTest(stats_dataset_test_base.StatsDatasetTestBase): optimization.assert_next( ["LatencyStats", "Map", "LatencyStats", "Prefetch", "LatencyStats"])).map(lambda x: x * x).prefetch(1).apply( - optimization.optimize(["latency_all_edges"])).apply( - stats_ops.set_stats_aggregator(stats_aggregator)) + stats_ops.set_stats_aggregator(stats_aggregator)).apply( + optimization.optimize(["latency_all_edges"])) iterator = dataset.make_initializable_iterator() get_next = iterator.get_next() summary_t = stats_aggregator.get_summary() |