pub struct InstanceNormBackwardPlan<T: Element, const N: usize> { /* private fields */ }Expand description
InstanceNorm BW plan — wraps GroupNormBackwardPlan.
Implementations§
Source§impl<T: Element, const N: usize> InstanceNormBackwardPlan<T, N>
impl<T: Element, const N: usize> InstanceNormBackwardPlan<T, N>
Sourcepub fn select(
stream: &Stream,
desc: &InstanceNormBackwardDescriptor<N>,
pref: PlanPreference,
) -> Result<Self>
pub fn select( stream: &Stream, desc: &InstanceNormBackwardDescriptor<N>, pref: PlanPreference, ) -> Result<Self>
Pick a kernel.
Sourcepub fn workspace_size(&self) -> usize
pub fn workspace_size(&self) -> usize
Workspace bytes.
Sourcepub fn precision_guarantee(&self) -> PrecisionGuarantee
pub fn precision_guarantee(&self) -> PrecisionGuarantee
Numerical guarantees.
Auto Trait Implementations§
impl<T, const N: usize> Freeze for InstanceNormBackwardPlan<T, N>
impl<T, const N: usize> RefUnwindSafe for InstanceNormBackwardPlan<T, N>where
T: RefUnwindSafe,
impl<T, const N: usize> Send for InstanceNormBackwardPlan<T, N>where
T: Send,
impl<T, const N: usize> Sync for InstanceNormBackwardPlan<T, N>where
T: Sync,
impl<T, const N: usize> Unpin for InstanceNormBackwardPlan<T, N>where
T: Unpin,
impl<T, const N: usize> UnsafeUnpin for InstanceNormBackwardPlan<T, N>
impl<T, const N: usize> UnwindSafe for InstanceNormBackwardPlan<T, N>where
T: UnwindSafe,
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