pub struct CandleAttentionOps;Trait Implementations§
Source§impl AttentionOps for CandleAttentionOps
impl AttentionOps for CandleAttentionOps
Source§fn attention(
&self,
q: &TensorRef,
k: &TensorRef,
v: &TensorRef,
params: &AttentionParams,
) -> Result<TensorRef>
fn attention( &self, q: &TensorRef, k: &TensorRef, v: &TensorRef, params: &AttentionParams, ) -> Result<TensorRef>
Standard multi-head / grouped-query attention. Read more
Source§fn paged_attention(
&self,
_q: &Arc<dyn TensorLike>,
_k_cache: &Arc<dyn TensorLike>,
_v_cache: &Arc<dyn TensorLike>,
_block_table: &[u32],
_params: &AttentionParams,
) -> Result<Arc<dyn TensorLike>, FerrumError>
fn paged_attention( &self, _q: &Arc<dyn TensorLike>, _k_cache: &Arc<dyn TensorLike>, _v_cache: &Arc<dyn TensorLike>, _block_table: &[u32], _params: &AttentionParams, ) -> Result<Arc<dyn TensorLike>, FerrumError>
Paged attention for KV-cache-based decode. Read more
Auto Trait Implementations§
impl Freeze for CandleAttentionOps
impl RefUnwindSafe for CandleAttentionOps
impl Send for CandleAttentionOps
impl Sync for CandleAttentionOps
impl Unpin for CandleAttentionOps
impl UnsafeUnpin for CandleAttentionOps
impl UnwindSafe for CandleAttentionOps
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more