Struct TensorArgs 

pub struct TensorArgs;

Type implementing MatmulArgs where all inputs and the output are materialized tensors.

Other types might implement MatmulArgs for fused matrix multiplication kernels.

Trait Implementations

impl Clone for TensorArgs

fn clone(&self) -> TensorArgs

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl MatmulArgs for TensorArgs

type Output<EO: Numeric> = View<Line<EO>, (u32, u32, u32), ReadWrite>

Type used for the output.

type Input<Lhs: Numeric, Rhs: Numeric, EO: Numeric> = TensorInputs<Lhs, Rhs, EO>

Type used for the input.

type State<Lhs: Numeric, Rhs: Numeric, EO: Numeric> = (View<Line<Lhs>, (u32, u32, u32)>, View<Line<Rhs>, (u32, u32, u32)>, CubeOption<View<Line<EO>, (u32, u32, u32)>>, View<Line<EO>, (u32, u32, u32), ReadWrite>)

Inner state that is used to create tensor inputs and tensor outputs .

fn init_state<Lhs: Numeric, Rhs: Numeric, EO: Numeric, G: GlobalConfig>( input: &Self::Input<Lhs, Rhs, EO>, output: &mut Self::Output<EO>, _config: G, ) -> Self::State<Lhs, Rhs, EO>

Init the state.

fn view_lhs<Lhs: Numeric, Rhs: Numeric, EO: Numeric>( state: &Self::State<Lhs, Rhs, EO>, ) -> View<Line<Lhs>, Coords3d>

fn view_rhs<Lhs: Numeric, Rhs: Numeric, EO: Numeric>( state: &Self::State<Lhs, Rhs, EO>, ) -> View<Line<Rhs>, Coords3d>

fn view_acc<Lhs: Numeric, Rhs: Numeric, EO: Numeric>( state: &Self::State<Lhs, Rhs, EO>, ) -> CubeOption<View<Line<EO>, Coords3d>>

fn view_out<Lhs: Numeric, Rhs: Numeric, EO: Numeric>( state: &mut Self::State<Lhs, Rhs, EO>, ) -> View<Line<EO>, Coords3d, ReadWrite>

fn __expand_init_state<Lhs: Numeric, Rhs: Numeric, EO: Numeric, G: GlobalConfig>( scope: &mut Scope, input: <Self::Input<Lhs, Rhs, EO> as CubeType>::ExpandType, output: <Self::Output<EO> as CubeType>::ExpandType, _config: G, ) -> <Self::State<Lhs, Rhs, EO> as CubeType>::ExpandType

fn __expand_view_lhs<Lhs: Numeric, Rhs: Numeric, EO: Numeric>( scope: &mut Scope, state: <Self::State<Lhs, Rhs, EO> as CubeType>::ExpandType, ) -> <View<Line<Lhs>, Coords3d> as CubeType>::ExpandType

fn __expand_view_rhs<Lhs: Numeric, Rhs: Numeric, EO: Numeric>( scope: &mut Scope, state: <Self::State<Lhs, Rhs, EO> as CubeType>::ExpandType, ) -> <View<Line<Rhs>, Coords3d> as CubeType>::ExpandType

fn __expand_view_acc<Lhs: Numeric, Rhs: Numeric, EO: Numeric>( scope: &mut Scope, state: <Self::State<Lhs, Rhs, EO> as CubeType>::ExpandType, ) -> <CubeOption<View<Line<EO>, Coords3d>> as CubeType>::ExpandType

fn __expand_view_out<Lhs: Numeric, Rhs: Numeric, EO: Numeric>( scope: &mut Scope, state: <Self::State<Lhs, Rhs, EO> as CubeType>::ExpandType, ) -> <View<Line<EO>, Coords3d, ReadWrite> as CubeType>::ExpandType