Searched defs:BatchDatasetOpTest (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | batch_dataset_op_test.cc | 27 class BatchDatasetOpTest : public DatasetOpsTestBase {}; class
154 ITERATOR_GET_NEXT_TEST_P(BatchDatasetOpTest, BatchDatasetParams, in ITERATOR_GET_NEXT_TEST_P() argument
163 TEST_F(BatchDatasetOpTest, DatasetTypeString) { in TEST_F() argument
172 TEST_F(BatchDatasetOpTest, DatasetOutputDtypes) { in TEST_F() argument
196 DATASET_OUTPUT_SHAPES_TEST_P(BatchDatasetOpTest, BatchDatasetParams, in DATASET_OUTPUT_SHAPES_TEST_P() argument
210 DATASET_CARDINALITY_TEST_P(BatchDatasetOpTest, BatchDatasetParams, in DATASET_CARDINALITY_TEST_P() argument
237 ITERATOR_OUTPUT_SHAPES_TEST_P(BatchDatasetOpTest, BatchDatasetParams, in ITERATOR_OUTPUT_SHAPES_TEST_P() argument
288 ITERATOR_SAVE_AND_RESTORE_TEST_P(BatchDatasetOpTest, BatchDatasetParams, in ITERATOR_SAVE_AND_RESTORE_TEST_P() argument
|