Struct RecurrentStateTensorSpec 

pub struct RecurrentStateTensorSpec {
    pub layer_index: usize,
    pub name: String,
    pub shape: Vec<usize>,
    pub dtype: DataType,
}

A single recurrent-state tensor owned by a model layer.

Fields

layer_index: usize

Layer index that owns this state tensor.

name: String

Backend/model-local state name, for example delta_state.

shape: Vec<usize>

Tensor shape excluding any request/batch slot dimension.

dtype: DataType

Storage dtype for this state tensor.

Implementations

impl RecurrentStateTensorSpec

pub fn new( layer_index: usize, name: impl Into<String>, shape: Vec<usize>, dtype: DataType, ) -> Self

pub fn checked_num_elements(&self) -> Option<usize>

pub fn num_elements(&self) -> usize

Trait Implementations

impl Clone for RecurrentStateTensorSpec

fn clone(&self) -> RecurrentStateTensorSpec

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for RecurrentStateTensorSpec

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for RecurrentStateTensorSpec

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Eq for RecurrentStateTensorSpec

impl PartialEq for RecurrentStateTensorSpec

fn eq(&self, other: &RecurrentStateTensorSpec) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl Serialize for RecurrentStateTensorSpec

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for RecurrentStateTensorSpec