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

  /external/tensorflow/tensorflow/python/tpu/
tpu_context.py 93 * self._internal_ctx.num_of_replicas_per_host)
100 replicas_consumed = self._internal_ctx.num_of_replicas_per_host
127 def num_of_replicas_per_host(self): member in class:TPUContext
131 'num_of_replicas_per_host is not supported for model_parallelism')
132 return self._internal_ctx.num_of_replicas_per_host
353 def num_of_replicas_per_host(self): member in class:_InternalTPUContext
694 num_of_replicas_per_host = self.num_of_replicas_per_host
695 host_id = replica_id / num_of_replicas_per_host
696 ordinal_id = replica_id % num_of_replicas_per_host
    [all...]
tpu_estimator.py 815 num_of_replicas_per_host = ctx.num_of_replicas_per_host
    [all...]

Completed in 54 milliseconds