Searched refs:_make_assignment_for_model (Results 1 – 1 of 1) sorted by relevance
236 def _make_assignment_for_model(self, cpu_model): member in TPUDistributionStrategy1414 self._tpu_assignment = strategy._make_assignment_for_model(cpu_model)