Struct PlacementOptimizer 

pub struct PlacementOptimizer { /* private fields */ }

Device placement optimizer

Implementations

impl PlacementOptimizer

pub fn new(strategy: PlacementStrategy, available_devices: Vec<Device>) -> Self

pub fn single_device(device: Device) -> Self

Create optimizer for single device

pub fn place(&self, graph: &EinsumGraph) -> PlacementPlan

Compute placement plan for a graph

pub fn place_with_schedule( &self, graph: &EinsumGraph, schedule: &ExecutionSchedule, ) -> PlacementPlan

Compute placement with an execution schedule

pub fn strategy(&self) -> PlacementStrategy

pub fn available_devices(&self) -> &[Device]