Home
last modified time | relevance | path

Searched defs:AutoShardDatasetParams (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dauto_shard_dataset_op_test.cc29 class AutoShardDatasetParams : public DatasetParams { class
32 AutoShardDatasetParams(T input_dataset_params, int64_t num_workers, in AutoShardDatasetParams() function in tensorflow::data::experimental::__anon268204ac0111::AutoShardDatasetParams
183 ITERATOR_GET_NEXT_TEST_P(AutoShardDatasetOpTest, AutoShardDatasetParams, in ITERATOR_GET_NEXT_TEST_P() argument