Searched refs:_next_instance_key (Results 1 – 1 of 1) sorted by relevance
295 def _next_instance_key(self): member in CollectiveReplicaLauncher316 self._next_instance_key, tensor_spec.TensorSpec([], dtypes.int32))353 instance_key = self._next_instance_key()378 instance_key = self._next_instance_key()