Struct TensorMapArgs 

pub struct TensorMapArgs;

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

Other types might implement MatmulArgs for fused matrix multiplication kernels.

Trait Implementations

impl Clone for TensorMapArgs

fn clone(&self) -> TensorMapArgs

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl MatmulArgs for TensorMapArgs

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

Type used for the input.

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

Type used for the output.

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