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

Update test post forward compatibility horizon update

Browse Source 
PiperOrigin-RevId: 299589163
Change-Id: Id407bee5405d45f58b9faa74f1be772398800da1
This commit is contained in:
Jacques Pienaar 2020-03-07 13:42:40 -08:00 committed by TensorFlower Gardener
parent d5e1a27a75
commit 022e00f942
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/python/data/experimental/kernel_tests/stats_dataset_ops_test.py
View File

@ -421,7 +421,7 @@ class FeatureStatsDatasetTest(
num_output = total_records // batch_size + 1
self.parallelCallsStats(
dataset_fn, {"ParseExampleDataset"},
dataset_fn, {"ParseExampleDatasetV2"},
num_output,
check_elements=False)
@ -439,19 +439,19 @@ class FeatureStatsDatasetTest(
handle = self.getHandle(aggregator)
self.assertStatisticsHasCount(
handle,
self.regexForNodeName("record_stats::ParseExampleDataset",
self.regexForNodeName("record_stats::ParseExampleDatasetV2",
"features_count"), total_records)
self.assertStatisticsHasCount(
handle,
self.regexForNodeName("record_stats::ParseExampleDataset",
self.regexForNodeName("record_stats::ParseExampleDatasetV2",
"feature_values_count"), total_records)
self.assertStatisticsHasSum(
handle,
self.regexForNodeName("record_stats::ParseExampleDataset",
self.regexForNodeName("record_stats::ParseExampleDatasetV2",
"features_count"), total_records * 4)
self.assertStatisticsHasSum(
handle,
self.regexForNodeName("record_stats::ParseExampleDataset",
self.regexForNodeName("record_stats::ParseExampleDatasetV2",
"feature_values_count"),
self._sum_keywords(1) * num_epochs + 3 * total_records)