Searched refs:RoundRobinTaskRunner (Results 1 – 3 of 3) sorted by relevance
98 out = absl::make_unique<RoundRobinTaskRunner>(std::move(iterator), in Create()124 RoundRobinTaskRunner::RoundRobinTaskRunner( in RoundRobinTaskRunner() function in tensorflow::data::RoundRobinTaskRunner133 Status RoundRobinTaskRunner::ValidateRequest(const GetElementRequest& req) { in ValidateRequest()147 Status RoundRobinTaskRunner::PrepareFullRound(int64 wait_us) in PrepareFullRound()157 Status RoundRobinTaskRunner::PreparePartialRound() in PreparePartialRound()175 Status RoundRobinTaskRunner::PrepareRound(const GetElementRequest& req) { in PrepareRound()210 Status RoundRobinTaskRunner::GetNext(const GetElementRequest& req, in GetNext()
111 RoundRobinTaskRunner runner(absl::make_unique<TestTaskIterator>(elements), in TEST_P()152 TEST(RoundRobinTaskRunner, ConsumeParallelPartialRound) { in TEST() argument164 RoundRobinTaskRunner runner(absl::make_unique<TestTaskIterator>(elements), in TEST()
131 class RoundRobinTaskRunner : public TaskRunner {133 RoundRobinTaskRunner(std::unique_ptr<TaskIterator> iterator,