HomeSort by relevance Sort by last modified time
    Searched refs:model_parallelism_enabled (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/tpu/
tpu_context.py 129 if self._internal_ctx.model_parallelism_enabled:
274 query_topology=self.model_parallelism_enabled))
330 def model_parallelism_enabled(self): member in class:_InternalTPUContext
355 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:
tpu_estimator.py 469 self._should_initialize_tpu = not ctx.model_parallelism_enabled
    [all...]

Completed in 469 milliseconds