pub struct ParallelBranchAttention { /* private fields */ }Implementations§
Source§impl ParallelBranchAttention
impl ParallelBranchAttention
pub fn new(config: ParallelBranchConfig) -> Self
Trait Implementations§
Source§impl DagAttentionMechanism for ParallelBranchAttention
impl DagAttentionMechanism for ParallelBranchAttention
Source§fn forward(&self, dag: &QueryDag) -> Result<AttentionScores, AttentionError>
fn forward(&self, dag: &QueryDag) -> Result<AttentionScores, AttentionError>
Compute attention scores for the given DAG
Source§fn complexity(&self) -> &'static str
fn complexity(&self) -> &'static str
Get computational complexity as a string
Auto Trait Implementations§
impl Freeze for ParallelBranchAttention
impl RefUnwindSafe for ParallelBranchAttention
impl Send for ParallelBranchAttention
impl Sync for ParallelBranchAttention
impl Unpin for ParallelBranchAttention
impl UnsafeUnpin for ParallelBranchAttention
impl UnwindSafe for ParallelBranchAttention
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