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 Булево. Используется ли в XLA SPMD-разделитель вместо MPMD-разделителя, когда запрошена разбиение компилятором?

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

Атрибуты
`use_spmd_for_xla_partitioning`	Булево. Используется ли в XLA SPMD-разделитель вместо MPMD-разделителя, когда запрошена разбиение компилятором?
`enable_xla_dynamic_padder`	Булево. Включить инфраструктуру динамического заполнителя XLA для обработки входных данных с динамическими формами внутри XLA. По умолчанию значение True. Отключение может привести к ошибкам корректности при входе с динамическими формами, поскольку 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/versions/r2.9/api_docs/python/tf/tpu/XLAOptions