balance_partitions

Function balance_partitions 

Source
pub fn balance_partitions(
    tensor: &Tensor,
    num_partitions: usize,
    initial_partitioning: &[usize],
    communication_scheme: CommunicationScheme,
    memory_limit: Option<f64>,
) -> (Vec<usize>, f64)
Expand description

Balances partitions using a genetic algorithm. Finds the partitioning that reduces the total contraction cost.