Struct VariableProfileTableBuilder

pub struct VariableProfileTableBuilder { /* private fields */ }

Builder for VariableProfileTable.

Implementations

impl VariableProfileTableBuilder

pub fn new() -> Result<Self, Error>

Create a builder.

let vpt_builder = VariableProfileTableBuilder::new()?;

pub fn add_input<T>(&mut self, name: &str, dims: &[usize]) -> Result<(), Error>

where T: Dtype,

Register a variable as input.

vpt_builder.add_input::<f32>("input", &[2, 3])?;

pub fn add_output(&mut self, name: &str) -> Result<(), Error>

Register a variable as output.

vpt_builder.add_output("fc2")?;

pub fn build( self, model_data: &ModelData, ) -> Result<VariableProfileTable, Error>

Build a VariableProfileTable using a ModelData.

let vpt = vpt_builder.build(&model_data)?;

Trait Implementations

impl Drop for VariableProfileTableBuilder

fn drop(&mut self)

Executes the destructor for this type.