Struct Plan 

pub struct Plan { /* private fields */ }

A reusable index map and execution description, independent of buffer addresses.

Implementations

impl Plan

pub fn new(operation: OperationDescriptor) -> Result<Self>

Resolve free, broadcast, diagonal and reduction axes without executing a kernel.

pub fn operation(&self) -> &OperationDescriptor

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

pub fn reduction_elements(&self) -> usize

pub fn execute<R: Runtime>( &self, inputs: &[&RudaTensor<R>], scalars: &[f64], ) -> Result<RudaTensor<R>>

Allocate D and enqueue the operation. All inputs remain unchanged.

pub fn execute_into<R: Runtime>( &self, inputs: &[&RudaTensor<R>], output: RudaTensor<R>, scalars: &[f64], ) -> Result<RudaTensor<R>>

Write into an exclusively owned D buffer and return it. D cannot alias any input.

Trait Implementations

impl Clone for Plan

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Plan

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

Formats the value using the given formatter.