tf.fixed_size_partitioner(num_shards, axis=0)

tf.fixed_size_partitioner(num_shards, axis=0)

See the guide: Variables > Variable Partitioners for Sharding

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

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, get_variable, and get_partitioned_variable_list.

Defined in tensorflow/python/ops/partitioned_variables.py.