Struct AutoLayoutOptimizer 

pub struct AutoLayoutOptimizer { /* private fields */ }

Automatic layout optimization pass for computation graphs

Implementations

impl AutoLayoutOptimizer

pub fn new() -> Self

Create a new automatic layout optimizer

pub fn register_tensor(&mut self, tensor_id: usize, layout: DataLayout)

Register a tensor with its initial layout

pub fn optimize_for_operation<T>( &mut self, tensors: &mut [&mut Tensor<T>], tensor_ids: &[usize], op_type: OperationType, device: &Device, ) -> Result<()>

where T: Clone + Default + Zero + One + Send + Sync + 'static + Pod + Zeroable,

Optimize layout for a specific operation

pub fn get_layout(&self, tensor_id: usize) -> Option<DataLayout>

Get the current layout of a tensor

pub fn total_cost(&self) -> f32

Get the total conversion cost incurred

pub fn reset(&mut self)

Reset the optimizer state

Trait Implementations

impl Default for AutoLayoutOptimizer

fn default() -> Self

Returns the “default value” for a type.