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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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>> std::vector<CardinalityTestCase<MapAndBatchDatasetParams>>
CardinalityTestCases() { CardinalityTestCases() {
return {{/*dataset_params=*/MapAndBatchDatasetParams1(), return {{/*dataset_params=*/MapAndBatchDatasetParams1(),
/*expected_cardinality=*/-2}, /*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/MapAndBatchDatasetParams2(), {/*dataset_params=*/MapAndBatchDatasetParams2(),
/*expected_cardinality=*/2}, /*expected_cardinality=*/2},
{/*dataset_params=*/MapAndBatchDatasetParams3(), {/*dataset_params=*/MapAndBatchDatasetParams3(),
/*expected_cardinality=*/3}, /*expected_cardinality=*/3},
{/*dataset_params=*/MapAndBatchDatasetParams4(), {/*dataset_params=*/MapAndBatchDatasetParams4(),
/*expected_cardinality=*/-2}, /*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/MapAndBatchDatasetParams5(), {/*dataset_params=*/MapAndBatchDatasetParams5(),
/*expected_cardinality=*/2}, /*expected_cardinality=*/2},
{/*dataset_params=*/MapAndBatchDatasetParams6(), {/*dataset_params=*/MapAndBatchDatasetParams6(),
/*expected_cardinality=*/-2}}; /*expected_cardinality=*/kUnknownCardinality}};
} }
DATASET_CARDINALITY_TEST_P(MapAndBatchDatasetOpTest, MapAndBatchDatasetParams, 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(), return {{/*dataset_params=*/MapDatasetParams1(),
/*expected_cardinality=*/4}, /*expected_cardinality=*/4},
{/*dataset_params=*/MapDatasetParams2(), {/*dataset_params=*/MapDatasetParams2(),
/*expected_cardinality=*/-2}, /*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/MapDatasetParams3(), {/*dataset_params=*/MapDatasetParams3(),
/*expected_cardinality=*/4}}; /*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>> std::vector<CardinalityTestCase<ParallelMapDatasetParams>>
CardinalityTestCases() { CardinalityTestCases() {
return {{/*dataset_params=*/ParallelMapDatasetParams1(), return {{/*dataset_params=*/ParallelMapDatasetParams1(),
/*expected_cardinality=*/-2}, /*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/ParallelMapDatasetParams2(), {/*dataset_params=*/ParallelMapDatasetParams2(),
/*expected_cardinality=*/4}, /*expected_cardinality=*/4},
{/*dataset_params=*/ParallelMapDatasetParams3(), {/*dataset_params=*/ParallelMapDatasetParams3(),
/*expected_cardinality=*/-2}, /*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/ParallelMapDatasetParams4(), {/*dataset_params=*/ParallelMapDatasetParams4(),
/*expected_cardinality=*/-2}, /*expected_cardinality=*/kUnknownCardinality},
{/*dataset_params=*/ParallelMapDatasetParams5(), {/*dataset_params=*/ParallelMapDatasetParams5(),
/*expected_cardinality=*/4}, /*expected_cardinality=*/4},
{/*dataset_params=*/ParallelMapDatasetParams6(), {/*dataset_params=*/ParallelMapDatasetParams6(),
/*expected_cardinality=*/-2}}; /*expected_cardinality=*/kUnknownCardinality}};
} }
DATASET_CARDINALITY_TEST_P(ParallelMapDatasetOpTest, ParallelMapDatasetParams, DATASET_CARDINALITY_TEST_P(ParallelMapDatasetOpTest, ParallelMapDatasetParams,