Searched refs:PaddedBatchDatasetParams (Results 1 – 1 of 1) sorted by relevance
25 class PaddedBatchDatasetParams : public DatasetParams { class28 PaddedBatchDatasetParams(T input_dataset_params, int64_t batch_size, in PaddedBatchDatasetParams() function in tensorflow::data::__anone4c89dd70111::PaddedBatchDatasetParams105 PaddedBatchDatasetParams PaddedBatchDatasetParams1() { in PaddedBatchDatasetParams1()110 return PaddedBatchDatasetParams( in PaddedBatchDatasetParams1()124 PaddedBatchDatasetParams PaddedBatchDatasetParams2() { in PaddedBatchDatasetParams2()138 return PaddedBatchDatasetParams( in PaddedBatchDatasetParams2()152 PaddedBatchDatasetParams PaddedBatchDatasetParams3() { in PaddedBatchDatasetParams3()166 return PaddedBatchDatasetParams( in PaddedBatchDatasetParams3()182 PaddedBatchDatasetParams PaddedBatchDatasetParams4() { in PaddedBatchDatasetParams4()196 return PaddedBatchDatasetParams( in PaddedBatchDatasetParams4()[all …]