Searched refs:_get_indexed_slices (Results 1 – 1 of 1) sorted by relevance
638 def _get_indexed_slices(self, member in CollectiveAllReduceTest691 per_replica = self._get_indexed_slices(devices,715 per_replica_on_cpu = self._get_indexed_slices(