tf.tpu.initialize_system
Инициализирует распределённую систему TPU для использования с TensorFlow.
tf.tpu.initialize_system(
embedding_config=None, job=None
)
| Аргументы | |
|---|---|
embedding_config | Если не None, то TPUEmbeddingConfiguration протокол, описывающий желаемую конфигурацию таблиц поиска встраивания аппаратного обеспечения. Если embedding_config равно None, то нельзя использовать аппаратные встраивания. |
job | Задача (XXX в спецификации устройства TensorFlow /job:XXX), содержащая устройства TPU, которые будут инициализированы. Если job=None, предполагается, что в TensorFlow flock существует только одна задача, и будет возвращена ошибка, если это предположение неверно. |
| Возвращает | |
|---|---|
Сериализованный TopologyProto, описывающий систему TPU. Примечание: топология должна быть оценена с помощью Session.run перед её использованием. |
© 2020 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/versions/r1.15/api_docs/python/tf/tpu/initialize_system