TensorFlow 2.0 Beta is available Learn more

tf.fixed_size_partitioner

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

Aliases:

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

Defined in python/ops/partitioned_variables.py.

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.