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

use kUnknownCardinality in test

Browse Source
This commit is contained in:
zilinzhu 2020-06-07 08:46:15 +08:00
parent 8b19be016a
commit 477203e325
3 changed files with 8 additions and 8 deletions

6
tensorflow/core/kernels/data/experimental/map_and_batch_dataset_op_test.cc
View File

@ -316,17 +316,17 @@ DATASET_OUTPUT_SHAPES_TEST_P(MapAndBatchDatasetOpTest, MapAndBatchDatasetParams,
std::vector<CardinalityTestCase<MapAndBatchDatasetParams>>
CardinalityTestCases() {
return {{/*dataset_params=*/MapAndBatchDatasetParams1(),
/*expected_cardinality=*/-2},
/*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/MapAndBatchDatasetParams2(),
/*expected_cardinality=*/2},
{/*dataset_params=*/MapAndBatchDatasetParams3(),
/*expected_cardinality=*/3},
{/*dataset_params=*/MapAndBatchDatasetParams4(),
/*expected_cardinality=*/-2},
/*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/MapAndBatchDatasetParams5(),
/*expected_cardinality=*/2},
{/*dataset_params=*/MapAndBatchDatasetParams6(),
/*expected_cardinality=*/-2}};
/*expected_cardinality=*/kUnknownCardinality}};
}
DATASET_CARDINALITY_TEST_P(MapAndBatchDatasetOpTest, MapAndBatchDatasetParams,

2
tensorflow/core/kernels/data/map_dataset_op_test.cc
View File

@ -134,7 +134,7 @@ std::vector<CardinalityTestCase<MapDatasetParams>> CardinalityTestCases() {
return {{/*dataset_params=*/MapDatasetParams1(),
/*expected_cardinality=*/4},
{/*dataset_params=*/MapDatasetParams2(),
/*expected_cardinality=*/-2},
/*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/MapDatasetParams3(),
/*expected_cardinality=*/4}};
}

8
tensorflow/core/kernels/data/parallel_map_dataset_op_test.cc
View File

@ -329,17 +329,17 @@ TEST_F(ParallelMapDatasetOpTest, DatasetOutputShapes) {
std::vector<CardinalityTestCase<ParallelMapDatasetParams>>
CardinalityTestCases() {
return {{/*dataset_params=*/ParallelMapDatasetParams1(),
/*expected_cardinality=*/-2},
/*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/ParallelMapDatasetParams2(),
/*expected_cardinality=*/4},
{/*dataset_params=*/ParallelMapDatasetParams3(),
/*expected_cardinality=*/-2},
/*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/ParallelMapDatasetParams4(),
/*expected_cardinality=*/-2},
/*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/ParallelMapDatasetParams5(),
/*expected_cardinality=*/4},
{/*dataset_params=*/ParallelMapDatasetParams6(),
/*expected_cardinality=*/-2}};
/*expected_cardinality=*/kUnknownCardinality}};
}
DATASET_CARDINALITY_TEST_P(ParallelMapDatasetOpTest, ParallelMapDatasetParams,