hub / github.com/alibaba/nann / initialize_system_for_tpu_embedding

Function initialize_system_for_tpu_embedding

tensorflow/tensorflow/python/tpu/tpu.py:112–135 · view source on GitHub ↗

Initializes a distributed TPU Embedding system for use with TensorFlow. The following two are equivalent: 1. initialize_system() with embedding_config. 2. initialize_system() without embedding_config, then initialize_system_for_tpu_embedding(). initialize_system() should not be called

(embedding_config, job=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 3

_tpu_system_device_nameFunction · 0.85

SerializeToStringMethod · 0.45

deviceMethod · 0.45

Tested by

no test coverage detected