Skip to main content

TensorDispatch

Trait TensorDispatch 

Source
pub trait TensorDispatch: Send + 'static {
    // Required methods
    fn op_tag(&self) -> &'static str;
    fn dtype_tag(&self) -> &'static str;
    fn dispatch(self: Box<Self>, ctx: &TensorDispatchCtx);
    fn fail_mock(self: Box<Self>);
}
Expand description

TensorDispatch is owned by kernel::tensor (Phase 2 cuTENSOR).

Required Methods§

Source

fn op_tag(&self) -> &'static str

Source

fn dtype_tag(&self) -> &'static str

Source

fn dispatch(self: Box<Self>, ctx: &TensorDispatchCtx)

Source

fn fail_mock(self: Box<Self>)

Implementors§