tf.tpu.XLAOptions | TensorFlow v2.16.1 (original) (raw)
tf.tpu.XLAOptions
Stay organized with collections Save and categorize content based on your preferences.
XLA compilation options.
View aliases
Compat aliases for migration
SeeMigration guide for more details.
tf.tpu.XLAOptions(
use_spmd_for_xla_partitioning=True, enable_xla_dynamic_padder=True
)
| Attributes | |
|---|---|
| use_spmd_for_xla_partitioning | Boolean. Whether to use XLA's SPMD partitioner instead of MPMD partitioner when compiler partitioning is requested. |
| enable_xla_dynamic_padder | Boolean. Whether to enable XLA dynamic padder infrastructure to handle dynamic shapes inputs inside XLA. True by default. Disabling this may cause correctness issues with dynamic shapes inputs, as XLA will just assume the inputs are with padded shapes. However users can optionally set it to False to improve device time if masking is already handled in the user side. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.