tf.tpu.XLAOptions

Параметры компиляции XLA.

Просмотреть псевдонимы

Псевдонимы совместимости для миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.tpu.XLAOptions

tf.tpu.XLAOptions(
    use_spmd_for_xla_partitioning=True, enable_xla_dynamic_padder=True
)

Атрибуты

use_spmd_for_xla_partitioning Логическое значение. Использовать ли разделитель SPMD XLA вместо разделителя MPMD при запросе разбиения компилятором.

enable_xla_dynamic_padder Логическое значение. Включать ли инфраструктуру динамического дополнения XLA для обработки входных данных с динамическими формами внутри XLA. По умолчанию включено. Отключение может привести к проблемам с корректностью при работе с входными данными с динамическими формами, поскольку XLA будет считать, что входные данные уже дополнены до нужных форм. Однако пользователи могут установить значение False, чтобы сократить время выполнения на устройстве, если маскирование уже выполняется на стороне пользователя.

Атрибуты
`use_spmd_for_xla_partitioning`	Логическое значение. Использовать ли разделитель SPMD XLA вместо разделителя MPMD при запросе разбиения компилятором.
`enable_xla_dynamic_padder`	Логическое значение. Включать ли инфраструктуру динамического дополнения XLA для обработки входных данных с динамическими формами внутри XLA. По умолчанию включено. Отключение может привести к проблемам с корректностью при работе с входными данными с динамическими формами, поскольку XLA будет считать, что входные данные уже дополнены до нужных форм. Однако пользователи могут установить значение False, чтобы сократить время выполнения на устройстве, если маскирование уже выполняется на стороне пользователя.

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