Searched refs:next_step_id_ (Results 1 – 4 of 4) sorted by relevance
83 gks->next_step_id_ = NewRandomStepId(); in RefreshStepIdSequenceAsync()119 gks->next_step_id_ = NewRandomStepId(); in GetStepSequenceAsync()126 ss->set_next_step_id(gks->next_step_id_); in GetStepSequenceAsync()144 gks->next_step_id_ = ss.next_step_id(); in UpdateStepSequences()153 return it->second->next_step_id_; in NextStepId()162 if (step_id == it->second->next_step_id_) { in RetireStepId()163 it->second->next_step_id_ = (it->second->next_step_id_ + 1) & kStepIdMask; in RetireStepId()165 it->second->next_step_id_ = CollectiveExecutor::kInvalidId; in RetireStepId()
74 : graph_key_(k), next_step_id_(CollectiveExecutor::kInvalidId) {} in GraphKeySequence()77 int64 next_step_id_; member
327 int64 next_step_id_; variable
504 next_step_id_(1), in XlaCompiler()533 int64 XlaCompiler::NextStepId() { return next_step_id_++; } in NextStepId()