tf.data.experimental.AutoShardPolicy

View source on GitHub

Class AutoShardPolicy

Represents the type of auto-sharding we enable.

Aliases:

  • Class tf.compat.v1.data.experimental.AutoShardPolicy
  • Class tf.compat.v2.data.experimental.AutoShardPolicy

Please see the DistributeOptions.auto_shard_policy documentation for more information on each type of autosharding.

Class Members

  • AUTO = <AutoShardPolicy.AUTO: 0>
  • DATA = <AutoShardPolicy.DATA: 2>
  • FILE = <AutoShardPolicy.FILE: 1>
  • OFF = <AutoShardPolicy.OFF: -1>