Struct DeviceState 

pub struct DeviceState<B: NeuralBackend, L> { /* private fields */ }

Fully device-resident state with one concrete value per architecture layer.

Implementations

impl<B: NeuralBackend, L> DeviceState<B, L>

pub fn create<E>( layout: StateLayout, create: impl FnMut(usize, &LayerCachePolicy) -> Result<L, E>, ) -> Result<Self, E>

Realizes every layer through a backend-specific construction closure.

Trait Implementations

impl<B: NeuralBackend, L> AsMut<[L]> for DeviceState<B, L>

fn as_mut(&mut self) -> &mut [L]

Converts this type into a mutable reference of the (usually inferred) input type.

impl<B: NeuralBackend, L> AsRef<[L]> for DeviceState<B, L>

fn as_ref(&self) -> &[L]

Converts this type into a shared reference of the (usually inferred) input type.

impl<B: NeuralBackend, L: Clone> Clone for DeviceState<B, L>

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<B: Debug + NeuralBackend, L: Debug> Debug for DeviceState<B, L>

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

Formats the value using the given formatter.

impl<B, L> LayerRuntimeState<B> for DeviceState<B, L>

where B: NeuralBackend, L: RuntimeLayerState<B>,

type LayerState = L

Concrete monomorphized layer state used by the architecture.

fn layer(&mut self, layer: usize) -> Result<&mut Self::LayerState, StateError>

Mutably borrows one architecture-global layer state.

impl<B, L> ResettableRuntimeState<B> for DeviceState<B, L>

where B: NeuralBackend, L: ResettableRuntimeLayerState<B>,

fn reset_segment(&mut self, segment: &StateSegmentId) -> Result<(), StateError>

Resets every layer in exactly one declared state segment.

impl<B, L> RuntimeState<B> for DeviceState<B, L>

where B: NeuralBackend, L: RuntimeLayerState<B>,

type RetainedValues<'a> = <L as RuntimeLayerState<B>>::RetainedValues<'a> where Self: 'a, B::Tensor: 'a

Concrete iterator retaining native values for one execution unit.

fn layout(&self) -> &StateLayout

Returns the exact layout used to create this realization.

fn retained_values( &self, _ordinal: usize, address: ExecutionUnitAddress, ) -> Result<Self::RetainedValues<'_>, StateError>

Borrows tensors retained by one execution unit without cloning handles.