Struct RecurrentStateSpec 

pub struct RecurrentStateSpec {
    pub request_id: RequestId,
    pub num_layers: usize,
    pub tensors: Vec<RecurrentStateTensorSpec>,
    pub device: Device,
    pub max_batch_slots: usize,
}

Allocation request for recurrent state.

Fields

request_id: RequestId

Request ID this allocation is for.

num_layers: usize

Number of model layers that may carry recurrent state.

tensors: Vec<RecurrentStateTensorSpec>

Concrete state tensors to allocate.

device: Device

Target device.

max_batch_slots: usize

Number of request/batch slots reserved by this handle.

Implementations

impl RecurrentStateSpec

pub fn estimated_memory_bytes(&self) -> usize

Trait Implementations

impl Clone for RecurrentStateSpec

fn clone(&self) -> RecurrentStateSpec

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for RecurrentStateSpec

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for RecurrentStateSpec

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Eq for RecurrentStateSpec

impl PartialEq for RecurrentStateSpec

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

Equality operator ==.

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

Inequality operator !=.

impl Serialize for RecurrentStateSpec

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 RecurrentStateSpec