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

tf.compat.v1.fixed_size_partitioner

Partitioner to specify a fixed number of shards along given axis.

num_shards int, number of shards to partition variable.
axis int, axis to partition on.

A partition function usable as the partitioner argument to variable_scope and get_variable.