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

  /external/tensorflow/tensorflow/python/tpu/
tpu_context.py 625 num_cores_per_host = self._get_tpu_system_metadata().num_of_cores_per_host
626 if num_cores_per_replica > num_cores_per_host:
630 'num_cores_per_host. num_cores_per_replica: {}, '
631 'num_cores_per_host: {}'.format(num_cores_per_replica,
632 num_cores_per_host))
tpu_embedding.py 233 for _ in range(embedding.num_cores_per_host):
396 def num_cores_per_host(self): member in class:TPUEmbedding
    [all...]
tpu_estimator.py 739 num_cores_per_host = ctx.num_of_cores_per_host
741 for core_ordinal in range(num_cores_per_host):
    [all...]

Completed in 75 milliseconds