Searched defs:BatchDatasetParams (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | batch_dataset_op_test.cc | 154 ITERATOR_GET_NEXT_TEST_P(BatchDatasetOpTest, BatchDatasetParams, in ITERATOR_GET_NEXT_TEST_P() 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset_test_base.h | 225 BatchDatasetParams(T input_dataset_params, int64_t batch_size, in BatchDatasetParams() function
|