Struct SparsemaxBackwardPlan 

pub struct SparsemaxBackwardPlan<T: Element, const N: usize> { /* private fields */ }

Sparsemax backward plan.

Implementations

impl<T: Element, const N: usize> SparsemaxBackwardPlan<T, N>

pub fn select( _stream: &Stream, desc: &SparsemaxBackwardDescriptor<N>, _pref: PlanPreference, ) -> Result<Self>

Pick a kernel.

pub fn can_implement( &self, args: &SparsemaxBackwardArgs<'_, T, N>, ) -> Result<()>

Validate args.

pub fn workspace_size(&self) -> usize

Workspace size in bytes.

pub fn sku(&self) -> KernelSku

Kernel SKU identity.

pub fn precision_guarantee(&self) -> PrecisionGuarantee

Numerical guarantees.

pub fn run( &self, stream: &Stream, _workspace: Workspace<'_>, args: SparsemaxBackwardArgs<'_, T, N>, ) -> Result<()>

Launch.