Struct Context

pub struct Context<'a, H> {
    pub tensors: &'a mut HashMap<TensorId, TensorIr>,
    pub handles: &'a mut HandleContainer<H>,
    pub scalars: &'a mut HashMap<ScalarId, ScalarValue>,
}

The context contains the relative graph tensor mapping so that a relative tensor id can be mapped to an existing tensor that can be fetched and updated with the handle container.

It also contains all scalar values, which can change even for the same graph. They are sorted in the order in which they appear in the graph.

Fields

tensors: &'a mut HashMap<TensorId, TensorIr>

The tensor mapping where local tensor id points to the updated tensor representation.

handles: &'a mut HandleContainer<H>

Handle container to retrieve tensors based on their representation.

scalars: &'a mut HashMap<ScalarId, ScalarValue>

Scalars found in the graph in the order they appeared.

Implementations

impl<'a, H> Context<'a, H>

pub fn new( tensors: &'a mut HashMap<TensorId, TensorIr>, handles: &'a mut HandleContainer<H>, scalars: &'a mut HashMap<ScalarId, ScalarValue>, ) -> Self

Constructs a new Context.

impl<H: Clone> Context<'_, H>

pub fn fork(&self) -> ContextOwned<H>

Fork the context into an owned context.