View source on GitHub |
Represents options for distributed data processing.
tf.data.experimental.DistributeOptions()
You can set the distribution options of a dataset through the
experimental_distribute
property of tf.data.Options
; the property is
an instance of tf.data.experimental.DistributeOptions
.
options = tf.data.Options()
options.experimental_distribute.auto_shard_policy = AutoShardPolicy.OFF
dataset = dataset.with_options(options)
Attributes | |
---|---|
auto_shard_policy
|
The type of sharding to use. See tf.data.experimental.AutoShardPolicy for additional information.
|
num_devices
|
The number of devices attached to this input pipeline. This will be automatically set by MultiDeviceIterator .
|
Methods
__eq__
__eq__(
other
)
Return self==value.
__ne__
__ne__(
other
)
Return self!=value.