Struct OperationDescriptor 

pub struct OperationDescriptor { /* private fields */ }

A buffer-independent operation. Input order is the order passed to its constructor.

Implementations

impl OperationDescriptor

pub fn contraction( a: OperandDescriptor, b: OperandDescriptor, c: Option<OperandDescriptor>, output: TensorDescriptor, modes: &[Mode], compute: ComputeType, ) -> Result<Self>

D = alpha * sum(op(A) * op(B)) + beta * op(C). C, if supplied, follows A and B in the execution input list.

pub fn sum_product( inputs: Vec<OperandDescriptor>, c: Option<OperandDescriptor>, output: TensorDescriptor, modes: &[Mode], compute: ComputeType, ) -> Result<Self>

General multi-operand sum of products. Modes absent from D are summed.

pub fn reduction( a: OperandDescriptor, c: Option<OperandDescriptor>, output: TensorDescriptor, modes: &[Mode], reduction: ReductionOp, compute: ComputeType, ) -> Result<Self>

D = alpha * reduce(op(A)) + beta * op(C).

pub fn permutation( a: OperandDescriptor, output: TensorDescriptor, modes: &[Mode], compute: ComputeType, ) -> Result<Self>

D = alpha * op(A), with a physical copy into the specified output layout.

pub fn elementwise_binary( a: OperandDescriptor, b: OperandDescriptor, output: TensorDescriptor, modes: &[Mode], combine: BinaryOp, compute: ComputeType, ) -> Result<Self>

D = combine(alpha * op(A), beta * op(B)).

pub fn elementwise_trinary( inputs: [OperandDescriptor; 3], output: TensorDescriptor, modes: &[Mode], combine_ab: BinaryOp, combine_abc: BinaryOp, compute: ComputeType, ) -> Result<Self>

D = combine_abc(combine_ab(alpha * op(A), beta * op(B)), gamma * op(C)).

pub fn inputs(&self) -> &[OperandDescriptor]

pub fn output(&self) -> &TensorDescriptor

pub fn output_modes(&self) -> &[Mode]

pub fn compute_type(&self) -> ComputeType

pub fn scalar_count(&self) -> usize

Trait Implementations

impl Clone for OperationDescriptor

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 OperationDescriptor

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

Formats the value using the given formatter.

impl Eq for OperationDescriptor

impl PartialEq for OperationDescriptor

fn eq(&self, other: &Self) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for OperationDescriptor