tf.compat.v1.min_max_variable_partitioner

View source on GitHub

Partitioner to allocate minimum size per slice.

tf.compat.v1.min_max_variable_partitioner(
    max_partitions=1, axis=0, min_slice_size=(256 << 10),
    bytes_per_string_element=16
)

Returns a partitioner that partitions the variable of given shape and dtype such that each partition has a minimum of min_slice_size slice of the variable. The maximum number of such partitions (upper bound) is given by max_partitions.

Args:

Returns:

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