Run options for strategy.run
.
tf.distribute.RunOptions( experimental_enable_dynamic_batch_size=True, experimental_bucketizing_dynamic_shape=False )
This can be used to hold some strategy specific configs.
Attributes | |
---|---|
experimental_enable_dynamic_batch_size | Boolean. Only applies to TPUStrategy. Default to True. If True, TPUStrategy will enable dynamic padder to support dynamic batch size for the inputs. Otherwise only static shape inputs are allowed. |
experimental_bucketizing_dynamic_shape | Boolean. Only applies to TPUStrategy. Default to False. If True, TPUStrategy will automatic bucketize inputs passed into run if the input shape is dynamic. This is a performance optimization to reduce XLA recompilation, which should not have impact on correctness. |
© 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/distribute/RunOptions