hub / github.com/alibaba/nann / initialize_tpu_system

Function initialize_tpu_system

tensorflow/tensorflow/python/tpu/tpu_strategy_util.py:40–118 · view source on GitHub ↗

Initialize the TPU devices. Args: cluster_resolver: A tf.distribute.cluster_resolver.TPUClusterResolver, which provides information about the TPU cluster. Returns: The tf.tpu.Topology object for the topology of the TPU cluster. Raises: RuntimeError: If no TPU devices foun

(cluster_resolver=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 15

get_job_nameMethod · 0.95

masterMethod · 0.95

cluster_specMethod · 0.95

TPUClusterResolverClass · 0.90

_tpu_init_fnFunction · 0.85

executing_eagerlyMethod · 0.80

from_stringMethod · 0.80

infoMethod · 0.80

_clear_cachesMethod · 0.80

as_cluster_defMethod · 0.80

contextMethod · 0.45

formatMethod · 0.45

Tested by

no test coverage detected