Struct GatherPlan 

pub struct GatherPlan { /* private fields */ }

A compiled gather traversal for one value layout, index layout, and output layout.

Implementations

impl GatherPlan

pub fn compile( operand_dims: &[usize], operand_strides: &[isize], index_dims: &[usize], index_strides: &[isize], dest_dims: &[usize], dest_strides: &[isize], spec: GatherSpec, ) -> Result<Self>

Compile a gather plan for fixed operand, index, and destination layouts.

pub fn spec(&self) -> &GatherSpec

pub fn dest_dims(&self) -> &[usize]

pub fn execute<T, I>( &self, dest: &mut RawStridedMut<'_, T>, operand: &RawStridedRef<'_, T>, start_indices: &RawStridedRef<'_, I>, ) -> Result<()>

where T: Copy + MaybeSendSync, I: GatherIndex,

Execute the prepared gather traversal.

Trait Implementations

impl Clone for GatherPlan

fn clone(&self) -> GatherPlan

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for GatherPlan

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.