pub enum ConvTranspose2dStrategy {
Direct,
Autotune,
Gemm,
}Expand description
The strategy to be used when launching a conv_transpose kernel.
Variants§
Direct
A simple direct convolution.
Autotune
Available on crate feature
autotune only.Using autotune to choose the best kernel based on runtime information.
Gemm
GEMM (im2col) based implementation of convolution. Significantly increased memory usage.
Trait Implementations§
Auto Trait Implementations§
impl Freeze for ConvTranspose2dStrategy
impl RefUnwindSafe for ConvTranspose2dStrategy
impl Send for ConvTranspose2dStrategy
impl Sync for ConvTranspose2dStrategy
impl Unpin for ConvTranspose2dStrategy
impl UnwindSafe for ConvTranspose2dStrategy
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