pub struct KernelDispatcher;
Expand description
Kernel dispatcher for automatic algorithm selection
Implementations§
Source§impl KernelDispatcher
impl KernelDispatcher
Sourcepub fn optimal_matmul(
a: &dyn SparseTensor,
b: &dyn SparseTensor,
) -> TorshResult<Box<dyn SparseTensor>>
pub fn optimal_matmul( a: &dyn SparseTensor, b: &dyn SparseTensor, ) -> TorshResult<Box<dyn SparseTensor>>
Choose optimal matrix multiplication algorithm based on matrix characteristics
Auto Trait Implementations§
impl Freeze for KernelDispatcher
impl RefUnwindSafe for KernelDispatcher
impl Send for KernelDispatcher
impl Sync for KernelDispatcher
impl Unpin for KernelDispatcher
impl UnwindSafe for KernelDispatcher
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> 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