Searched refs:NextAvailableTaskId (Results 1 – 4 of 4) sorted by relevance
200 int64 NextAvailableTaskId() const;
266 int64 task_id = state.NextAvailableTaskId(); in TEST()270 EXPECT_EQ(state.NextAvailableTaskId(), task_id + 1); in TEST()
348 int64 DispatcherState::NextAvailableTaskId() const { in NextAvailableTaskId() function in tensorflow::data::DispatcherState
589 int64 task_id = state_.NextAvailableTaskId(); in CreatePendingTask()607 int64 task_id = state_.NextAvailableTaskId(); in CreateTask()