W3cubDocs

/TensorFlow Python

tf.contrib.tpu.initialize_system

tf.contrib.tpu.initialize_system(
    embedding_config=None,
    job=None
)

Defined in tensorflow/contrib/tpu/python/tpu/tpu.py.

Initializes a distributed TPU system for use with TensorFlow.

Args:

  • embedding_config: If not None, an EmbeddingLayerConfiguration proto describing the desired configuration of the hardware embedding lookup tables. If embedding_config is None, no hardware embeddings can be used.
  • job: The job (the XXX in TensorFlow device specification /job:XXX) that contains the TPU devices that will be initialized. If job=None it is assumed there is only one job in the TensorFlow flock, and an error will be returned if this assumption does not hold.

Returns:

A serialized TopologyProto that describes the TPU system. Note: the topology must be evaluated using Session.run before it can be used.

© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/tpu/initialize_system