#include <arm_gemm.hpp>

Collaboration diagram for GemmArgs:

Public Member Functions
	GemmArgs (const CPUInfo ci, unsigned int M, unsigned int N, unsigned int K, unsigned int Ksections, unsigned int nbatches, unsigned int nmulti, bool indirect_input, Activation act, const int maxthreads, bool fixed_format=false, bool fast_mode=false, const GemmConfig cfg=nullptr)

Data Fields
const CPUInfo *	_ci

unsigned int	_Msize

unsigned int	_Nsize

unsigned int	_Ksize

unsigned int	_Ksections

unsigned int	_nbatches

unsigned int	_nmulti

bool	_indirect_input

Activation	_act

int	_maxthreads

bool	_fixed_format

bool	_fast_mode

const GemmConfig *	_cfg

Detailed Description

Definition at line 142 of file arm_gemm.hpp.

Constructor & Destructor Documentation

◆ GemmArgs()

GemmArgs	(	const CPUInfo *	ci,
		unsigned int	M,
		unsigned int	N,
		unsigned int	K,
		unsigned int	Ksections,
		unsigned int	nbatches,
		unsigned int	nmulti,
		bool	indirect_input,
		Activation	act,
		const int	maxthreads,
		bool	fixed_format = `false`,
		bool	fast_mode = `false`,
		const GemmConfig *	cfg = `nullptr`
	)

inline

Definition at line 159 of file arm_gemm.hpp.

         : _ci(ci), _Msize(M), _Nsize(N), _Ksize(K), _Ksections(Ksections), _nbatches(nbatches), _nmulti(nmulti), _indirect_input(indirect_input), _act(act), _maxthreads(maxthreads),
           _fixed_format(fixed_format), _fast_mode(fast_mode), _cfg(cfg)
     {
     }

Field Documentation

◆ _act

Activation _act

Definition at line 153 of file arm_gemm.hpp.

Referenced by GemvPretransposed< strategy, To, Tr, OutputStage >::execute(), and GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::execute().

◆ _cfg

const GemmConfig* _cfg

Definition at line 157 of file arm_gemm.hpp.

Referenced by GemmImplementation< Top, Tret, OutputStage >::do_is_supported(), GemmImplementation< Top, Tret, Nothing >::do_is_supported(), arm_gemm::find_implementation(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::GemmHybridIndirect(), GemmInterleavedPretransposed2d< strategy, To, Tr >::GemmInterleavedPretransposed2d(), GemvBatched< To, Tr >::GemvBatched(), and GemvPretransposed< strategy, To, Tr, OutputStage >::GemvPretransposed().

◆ _ci

const CPUInfo* _ci

Definition at line 145 of file arm_gemm.hpp.

Referenced by GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::estimate_cycles(), GemmInterleaved< strategy, To, Tr, OutputStage, MergeStep, FixedFormat, ForceThreadColumns >::estimate_cycles(), GemvPretransposed< strategy, To, Tr, OutputStage >::execute(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::execute(), GemvPretransposed< strategy, To, Tr, OutputStage >::pretranspose_B_array(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::pretranspose_B_array(), and QuantizeWrapper< To, Tr, Tgemm >::QuantizeWrapper().

◆ _fast_mode

bool _fast_mode

Definition at line 156 of file arm_gemm.hpp.

◆ _fixed_format

bool _fixed_format

Definition at line 155 of file arm_gemm.hpp.

Referenced by GemmImplementation< Top, Tret, OutputStage >::do_is_supported(), and GemmImplementation< Top, Tret, Nothing >::do_is_supported().

◆ _indirect_input

bool _indirect_input

Definition at line 152 of file arm_gemm.hpp.

Referenced by GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::execute(), and QuantizeWrapper< To, Tr, Tgemm >::QuantizeWrapper().

◆ _Ksections

unsigned int _Ksections

Definition at line 149 of file arm_gemm.hpp.

Referenced by GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::execute(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::pretranspose_B_array(), QuantizeWrapper< To, Tr, Tgemm >::QuantizeWrapper(), and GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::requantize_bias().

◆ _Ksize

unsigned int _Ksize

Definition at line 148 of file arm_gemm.hpp.

◆ _maxthreads

int _maxthreads

Definition at line 154 of file arm_gemm.hpp.

Referenced by GemmInterleavedPretransposed2d< strategy, To, Tr >::estimate_cycles(), GemmInterleaved< strategy, To, Tr, OutputStage, MergeStep, FixedFormat, ForceThreadColumns >::estimate_cycles(), QuantizeWrapper< To, Tr, Tgemm >::QuantizeWrapper(), and QuantizeWrapper< To, Tr, Tgemm >::set_nthreads().

◆ _Msize

unsigned int _Msize

Definition at line 146 of file arm_gemm.hpp.

Referenced by GemmHybrid< strategy, To, Tr >::estimate_cycles(), GemmInterleavedPretransposed2d< strategy, To, Tr >::estimate_cycles(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::estimate_cycles(), GemmInterleaved< strategy, To, Tr, OutputStage, MergeStep, FixedFormat, ForceThreadColumns >::estimate_cycles(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::execute(), GemvBatched< To, Tr >::GemvBatched(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::get_B_pretransposed_array_size(), and QuantizeWrapper< To, Tr, Tgemm >::QuantizeWrapper().

◆ _nbatches

unsigned int _nbatches

Definition at line 150 of file arm_gemm.hpp.

Referenced by GemmHybrid< strategy, To, Tr >::estimate_cycles(), GemmInterleavedPretransposed2d< strategy, To, Tr >::estimate_cycles(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::estimate_cycles(), GemmInterleaved< strategy, To, Tr, OutputStage, MergeStep, FixedFormat, ForceThreadColumns >::estimate_cycles(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::execute(), GemvBatched< To, Tr >::GemvBatched(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::get_B_pretransposed_array_size(), and QuantizeWrapper< To, Tr, Tgemm >::QuantizeWrapper().

◆ _nmulti

unsigned int _nmulti

Definition at line 151 of file arm_gemm.hpp.

◆ _Nsize

unsigned int _Nsize

Definition at line 147 of file arm_gemm.hpp.

The documentation for this struct was generated from the following file:

src/cpu/kernels/assembly/arm_gemm.hpp

Public Member Functions

Data Fields