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/api_docs/python/tf/tpu/XLAOptions