Searched defs:PaddedBatchDatasetParams (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | padded_batch_dataset_op_test.cc | 25 class PaddedBatchDatasetParams : public DatasetParams { class
28 PaddedBatchDatasetParams(T input_dataset_params, int64_t batch_size, in PaddedBatchDatasetParams() function in tensorflow::data::__anone4c89dd70111::PaddedBatchDatasetParams
528 ITERATOR_GET_NEXT_TEST_P(PaddedBatchDatasetOpTest, PaddedBatchDatasetParams, in ITERATOR_GET_NEXT_TEST_P() argument
564 DATASET_OUTPUT_DTYPES_TEST_P(PaddedBatchDatasetOpTest, PaddedBatchDatasetParams, in DATASET_OUTPUT_DTYPES_TEST_P() argument
585 DATASET_OUTPUT_SHAPES_TEST_P(PaddedBatchDatasetOpTest, PaddedBatchDatasetParams, in DATASET_OUTPUT_SHAPES_TEST_P() argument
606 DATASET_CARDINALITY_TEST_P(PaddedBatchDatasetOpTest, PaddedBatchDatasetParams, in DATASET_CARDINALITY_TEST_P() argument
628 PaddedBatchDatasetParams, in ITERATOR_OUTPUT_DTYPES_TEST_P() argument
650 PaddedBatchDatasetParams, in ITERATOR_OUTPUT_SHAPES_TEST_P() argument
710 PaddedBatchDatasetParams, in ITERATOR_SAVE_AND_RESTORE_TEST_P() argument
|