ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf.distribute.experimental.partitioners

Public API for tf.distribute.experimental.partitioners namespace.


class FixedShardsPartitioner: Partitioner that allocates a fixed number of shards.

class MaxSizePartitioner: Partitioner that keeps shards below max_shard_bytes.

class MinSizePartitioner: Partitioner that allocates a minimum size per shard.

class Partitioner: Partitioner base class: all partitiners inherit from this class.