Struct ShapeInference 

pub struct ShapeInference;

Implementations

impl ShapeInference

pub fn infer_output_shape( op: &TensorOp, inputs: &[&TensorTypeInfo], ) -> Result<Vec<TensorTypeInfo>, String>

pub fn compute_flops(op: &TensorOp, inputs: &[&TensorTypeInfo]) -> Option<u64>

pub fn compute_memory_bytes( op: &TensorOp, inputs: &[&TensorTypeInfo], ) -> Option<u64>

Trait Implementations

impl Debug for ShapeInference

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

Formats the value using the given formatter.