Searched defs:AutoShardDatasetParams (Results 1 – 1 of 1) sorted by relevance
29 class AutoShardDatasetParams : public DatasetParams { class32 AutoShardDatasetParams(T input_dataset_params, int64_t num_workers, in AutoShardDatasetParams() function in tensorflow::data::experimental::__anonba308f2c0111::AutoShardDatasetParams183 ITERATOR_GET_NEXT_TEST_P(AutoShardDatasetOpTest, AutoShardDatasetParams, in ITERATOR_GET_NEXT_TEST_P() argument