Searched refs:TasksForWorker (Results 1 – 4 of 4) sorted by relevance
285 TF_EXPECT_OK(state.TasksForWorker(worker_address, tasks)); in TEST()346 TF_EXPECT_OK(state.TasksForWorker(worker_address, tasks)); in TEST()365 TF_EXPECT_OK(state.TasksForWorker(worker_address_1, tasks)); in TEST()370 TF_EXPECT_OK(state.TasksForWorker(worker_address_2, tasks)); in TEST()381 TF_EXPECT_OK(state.TasksForWorker(worker_address, tasks)); in TEST()
209 Status TasksForWorker(const absl::string_view worker_address,
332 Status DispatcherState::TasksForWorker( in TasksForWorker() function in tensorflow::data::DispatcherState
213 Status s = state_.TasksForWorker(worker_address, correct_tasks); in WorkerHeartbeat()225 TF_RETURN_IF_ERROR(state_.TasksForWorker(worker_address, correct_tasks)); in WorkerHeartbeat()