Searched refs:_use_ordering_token (Results 1 – 1 of 1) sorted by relevance
272 if self._use_ordering_token():279 if control_input is not None and not self._use_ordering_token():288 def _use_ordering_token(self): member in CollectiveReplicaLauncher327 if self._use_ordering_token() and communication_hint == 'NCCL':333 return self._use_ordering_token()