Struct FusedKernel 

pub struct FusedKernel {
    pub id: String,
    pub name: String,
    pub pattern: FusionPattern,
    pub operations: Vec<String>,
    pub inputs: Vec<TensorInfo>,
    pub outputs: Vec<TensorInfo>,
    pub estimated_speedup: f64,
    pub memory_savings: usize,
    pub implementation: KernelImplementation,
}

Fused kernel representation

Fields

id: String

name: String

pattern: FusionPattern

operations: Vec<String>

inputs: Vec<TensorInfo>

outputs: Vec<TensorInfo>

estimated_speedup: f64

memory_savings: usize

implementation: KernelImplementation

Implementations

impl FusedKernel

pub fn new( id: String, name: String, pattern: FusionPattern, operations: Vec<String>, ) -> Self

pub fn with_inputs(self, inputs: Vec<TensorInfo>) -> Self

pub fn with_outputs(self, outputs: Vec<TensorInfo>) -> Self

pub fn with_speedup(self, speedup: f64) -> Self

pub fn with_memory_savings(self, savings: usize) -> Self

pub fn with_implementation(self, implementation: KernelImplementation) -> Self

Trait Implementations

impl Clone for FusedKernel

fn clone(&self) -> FusedKernel

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for FusedKernel

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.