Struct Executor 

pub struct Executor<B: Backend> { /* private fields */ }

Executes IrProgram graphs on the Shrew tensor runtime.

Implementations

impl<B: Backend> Executor<B>

pub fn new( program: IrProgram, device: B::Device, config: RuntimeConfig, ) -> Result<Self>

Create a new executor. Initializes all parameters.

pub fn program(&self) -> &IrProgram

Get the underlying IR program.

pub fn config(&self) -> &RuntimeConfig

Get a reference to the runtime config.

pub fn config_mut(&mut self) -> &mut RuntimeConfig

Get a mutable reference to the runtime config.

pub fn params(&self) -> &HashMap<(String, String), Tensor<B>>

Get all parameter tensors.

pub fn all_params(&self) -> Vec<Tensor<B>>

Get flattened parameter list (all graphs).

pub fn named_params(&self) -> Vec<(String, Tensor<B>)>

Get all parameters as (key, tensor) pairs, where key = "graph/param".

pub fn set_param_by_key(&mut self, key: &str, tensor: Tensor<B>) -> bool

Set a parameter by its "graph/param" key. Returns true if found.

pub fn device(&self) -> &B::Device

The device this executor is running on.

pub fn run( &self, graph_name: &str, inputs: &HashMap<String, Tensor<B>>, ) -> Result<ExecResult<B>>

Execute a named graph with given inputs.

pub fn update_params(&mut self, graph_name: &str, new_params: &[Tensor<B>])

Update parameters after an optimizer step.

pub fn graph_params(&self, graph_name: &str) -> Vec<Tensor<B>>

Collect parameters for a specific graph (for optimizer).