Skip to main content

partition_tensors

Function partition_tensors 

Source
pub fn partition_tensors<T: Operator>(
    operator: &T,
    input_tensors: &[Option<Tensor>],
    output_tensors: &[Option<Tensor>],
    num_partitions: usize,
) -> Result<Vec<TensorPartition>, SimError>
Expand description

Create partitions from full Tensors

This is a wrapper function to create TensorViews and then call create_partitions using those views.