tf.compat.v1.tpu.initialize_system
Инициализирует распределённую систему TPU для использования с TensorFlow.
tf.compat.v1.tpu.initialize_system(
embedding_config=None, job=None, compilation_failure_closes_chips=True
)
| Аргументы | |
|---|---|
embedding_config | Если не None, то протокол TPUEmbeddingConfiguration, описывающий желаемую конфигурацию таблиц поиска аппаратных вложений. Если embedding_config равно None, аппаратные вложения использовать нельзя. |
job | Задача (XXX в спецификации устройства TensorFlow /job:XXX), содержащая устройства TPU, которые будут инициализированы. Если job=None, предполагается, что в TensorFlow flock существует только одна задача, и будет возвращено сообщение об ошибке, если это предположение неверно. |
compilation_failure_closes_chips | Устанавливает конфигурацию, определяющую, нужно ли закрывать чипы TPU при ошибке компиляции. |
| Возвращаемое значение | |
|---|---|
Сериализованный 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/r2.3/api_docs/python/tf/compat/v1/tpu/initialize_system
