tf.compat.v1.fixed_size_partitioner

View source on GitHub

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

tf.compat.v1.fixed_size_partitioner(
    num_shards, axis=0
)

Args:

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

Returns:

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