Struct BlockSparseContraction 

pub struct BlockSparseContraction;

Block-sparse contraction: the A operand is block-sparse, B/C/D dense.

Implementations

impl BlockSparseContraction

pub unsafe fn new<'h>( handle: &'h Handle, a: &BlockSparseTensorDescriptor<'h>, modes_a: &[i32], b: &TensorDescriptor<'h>, modes_b: &[i32], c: &TensorDescriptor<'h>, modes_c: &[i32], d: &TensorDescriptor<'h>, modes_d: &[i32], compute_desc: *const c_void, ) -> Result<OperationDescriptor<'h>>

Safety

compute_desc must be null or a live cutensorComputeDescriptor_t.

Trait Implementations

impl Debug for BlockSparseContraction

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

Formats the value using the given formatter.