Searched refs:model_parallelism_enabled (Results 1 – 2 of 2) sorted by relevance
129 if self._internal_ctx.model_parallelism_enabled:274 query_topology=self.model_parallelism_enabled))330 def model_parallelism_enabled(self): member in _InternalTPUContext355 if self.model_parallelism_enabled:364 if self.model_parallelism_enabled:538 if self.model_parallelism_enabled:555 if self.model_parallelism_enabled:579 if self.model_parallelism_enabled:687 if self.model_parallelism_enabled:
469 self._should_initialize_tpu = not ctx.model_parallelism_enabled