Struct RecurrentTrackerWrapper 

pub struct RecurrentTrackerWrapper<Inner: InfererWrapper> { /* private fields */ }

A wrapper that adds recurrent state tracking to an inner model.

This is an alternative to using RecurrentTracker which allows separate state tracking from the inferer itself.

Implementations

impl<Inner: InfererWrapper> RecurrentTrackerWrapper<Inner>

pub fn wrap<T: Inferer>( inner: Inner, inferer: &T, ) -> Result<RecurrentTrackerWrapper<Inner>>

Wraps the provided inferer to automatically track any keys that are both inputs/outputs.

pub fn new<T: Inferer>( inner: Inner, inferer: &T, info: Vec<RecurrentInfo>, ) -> Result<Self>

Create a new recurrency tracker for the model.

Trait Implementations

impl<Inner: InfererWrapper> InfererWrapper for RecurrentTrackerWrapper<Inner>

fn invoke( &self, inferer: &dyn Inferer, batch: &mut ScratchPadView<'_>, ) -> Result<()>

Invokes the inner inferer, applying any additional logic before and after the call.

fn input_shapes<'a>( &'a self, _inferer: &'a dyn Inferer, ) -> &'a [(String, Vec<usize>)]

Returns the input shapes after this wrapper has been applied.

fn output_shapes<'a>( &'a self, _inferer: &'a dyn Inferer, ) -> &'a [(String, Vec<usize>)]

Returns the output shapes after this wrapper has been applied.

fn begin_agent(&self, inferer: &dyn Inferer, id: u64)

Called when starting inference for a new agent.

fn end_agent(&self, inferer: &dyn Inferer, id: u64)

Called when finishing inference for an agent.