Searched refs:next_step_id_ (Results 1 – 4 of 4) sorted by relevance
80 gks->next_step_id_ = NewRandomStepId(); in RefreshStepIdSequenceAsync()116 gks->next_step_id_ = NewRandomStepId(); in GetStepSequenceAsync()123 ss->set_next_step_id(gks->next_step_id_); in GetStepSequenceAsync()141 gks->next_step_id_ = ss.next_step_id(); in UpdateStepSequences()150 return it->second->next_step_id_; in NextStepId()159 if (step_id == it->second->next_step_id_) { in RetireStepId()160 it->second->next_step_id_ = (it->second->next_step_id_ + 1) & kStepIdMask; in RetireStepId()162 it->second->next_step_id_ = CollectiveExecutor::kInvalidId; in RetireStepId()
73 : graph_key_(k), next_step_id_(CollectiveExecutor::kInvalidId) {} in GraphKeySequence()76 int64 next_step_id_; member
454 int64 next_step_id_; variable
426 next_step_id_(1), in XlaCompiler()462 int64 XlaCompiler::NextStepId() { return next_step_id_++; } in NextStepId()