tf.distribute.experimental.partitioners.FixedShardsPartitioner

Partitioner that allocates a fixed number of shards.

Inherits From: Partitioner

Examples:

# standalone usage:
partitioner = FixedShardsPartitioner(num_shards=2)
partitions = partitioner(tf.TensorShape([10, 3]), tf.float32)