pub struct ScanPlan<T: Element, const N: usize> { /* private fields */ }Expand description
Single-axis scan forward plan — see the module-level docs for formulas, dtypes, workspace, and precision guarantees.
T: Element is the element type (f32 / f64 / f16 / bf16).
const N: usize is the tensor rank (1..=8).
Implementations§
Source§impl<T: Element, const N: usize> ScanPlan<T, N>
impl<T: Element, const N: usize> ScanPlan<T, N>
Sourcepub fn select(
_stream: &Stream,
desc: &ScanDescriptor<N>,
_pref: PlanPreference,
) -> Result<Self>
pub fn select( _stream: &Stream, desc: &ScanDescriptor<N>, _pref: PlanPreference, ) -> Result<Self>
Pick a kernel for desc. Validates scan_axis < N, the dtype
is in the wired FP family, and tensor rank ≤ 8. Returns
Error::Unsupported for cells outside the wired matrix.
Sourcepub fn can_implement(&self, args: &ScanArgs<'_, T, N>) -> Result<()>
pub fn can_implement(&self, args: &ScanArgs<'_, T, N>) -> Result<()>
Validate args.
Sourcepub fn workspace_size(&self) -> usize
pub fn workspace_size(&self) -> usize
Workspace size in bytes. Always zero.
Sourcepub fn precision_guarantee(&self) -> PrecisionGuarantee
pub fn precision_guarantee(&self) -> PrecisionGuarantee
Numerical guarantees for this plan’s kernel — deterministic, bit-stable on the same hardware, f32 accumulator for f16 / bf16 inputs (FP detour).
Auto Trait Implementations§
impl<T, const N: usize> Freeze for ScanPlan<T, N>
impl<T, const N: usize> RefUnwindSafe for ScanPlan<T, N>where
T: RefUnwindSafe,
impl<T, const N: usize> Send for ScanPlan<T, N>where
T: Send,
impl<T, const N: usize> Sync for ScanPlan<T, N>where
T: Sync,
impl<T, const N: usize> Unpin for ScanPlan<T, N>where
T: Unpin,
impl<T, const N: usize> UnsafeUnpin for ScanPlan<T, N>
impl<T, const N: usize> UnwindSafe for ScanPlan<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