Struct tract_linalg::frame::mmm::ScratchSpaceImpl
source · pub struct ScratchSpaceImpl<TI: LADatum> { /* private fields */ }
Implementations§
source§impl<TI: LADatum> ScratchSpaceImpl<TI>
impl<TI: LADatum> ScratchSpaceImpl<TI>
pub unsafe fn prepare<K: MatMatMulKer<TI>>( &mut self, m: usize, n: usize, specs: &[FusedSpec<'_>] ) -> TractResult<()>
pub unsafe fn run<K: MatMatMulKer<TI>>( &mut self, specs: &[FusedSpec<'_>], down: usize, right: usize )
pub unsafe fn for_valid_tile<K: MatMatMulKer<TI>>( &mut self, specs: &[FusedSpec<'_>], down: usize, right: usize )
pub unsafe fn for_border_tile<K: MatMatMulKer<TI>>( &mut self, specs: &[FusedSpec<'_>], down: usize, right: usize, m_remnant: usize, n_remnant: usize )
pub fn uspecs(&self) -> &[FusedKerSpec<TI>]
pub unsafe fn postprocess_tile<K: MatMatMulKer<TI>>(
&mut self,
specs: &[FusedSpec<'_>],
down: usize,
right: usize,
m_remnant: usize,
n_remnant: usize
)where
TI: LADatum,
Trait Implementations§
source§impl<TI: LADatum> Default for ScratchSpaceImpl<TI>
impl<TI: LADatum> Default for ScratchSpaceImpl<TI>
source§impl<TI: LADatum> Drop for ScratchSpaceImpl<TI>
impl<TI: LADatum> Drop for ScratchSpaceImpl<TI>
impl<TI: LADatum> ScratchSpace for ScratchSpaceImpl<TI>
impl<TI: LADatum> Send for ScratchSpaceImpl<TI>
Auto Trait Implementations§
impl<TI> Freeze for ScratchSpaceImpl<TI>
impl<TI> RefUnwindSafe for ScratchSpaceImpl<TI>where
TI: RefUnwindSafe,
impl<TI> !Sync for ScratchSpaceImpl<TI>
impl<TI> Unpin for ScratchSpaceImpl<TI>where
TI: Unpin,
impl<TI> UnwindSafe for ScratchSpaceImpl<TI>where
TI: UnwindSafe + RefUnwindSafe,
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> Downcast for Twhere
T: Any,
impl<T> Downcast for Twhere
T: Any,
source§fn into_any(self: Box<T>) -> Box<dyn Any>
fn into_any(self: Box<T>) -> Box<dyn Any>
Convert
Box<dyn Trait>
(where Trait: Downcast
) to Box<dyn Any>
. Box<dyn Any>
can
then be further downcast
into Box<ConcreteType>
where ConcreteType
implements Trait
.source§fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
Convert
Rc<Trait>
(where Trait: Downcast
) to Rc<Any>
. Rc<Any>
can then be
further downcast
into Rc<ConcreteType>
where ConcreteType
implements Trait
.source§fn as_any(&self) -> &(dyn Any + 'static)
fn as_any(&self) -> &(dyn Any + 'static)
Convert
&Trait
(where Trait: Downcast
) to &Any
. This is needed since Rust cannot
generate &Any
’s vtable from &Trait
’s.source§fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
Convert
&mut Trait
(where Trait: Downcast
) to &Any
. This is needed since Rust cannot
generate &mut Any
’s vtable from &mut Trait
’s.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