#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 139 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 156 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 150 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 154 of file arm_gemm.hpp.

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

◆ _ci

const CPUInfo* _ci

Definition at line 142 of file arm_gemm.hpp.

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

◆ _fast_mode

bool _fast_mode

Definition at line 153 of file arm_gemm.hpp.

◆ _fixed_format

bool _fixed_format

Definition at line 152 of file arm_gemm.hpp.

◆ _indirect_input

bool _indirect_input

Definition at line 149 of file arm_gemm.hpp.

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

◆ _Ksections

unsigned int _Ksections

Definition at line 146 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_part(), and GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::requantize_bias().

◆ _Ksize

unsigned int _Ksize

Definition at line 145 of file arm_gemm.hpp.

◆ _maxthreads

int _maxthreads

Definition at line 151 of file arm_gemm.hpp.

Referenced by QuantizeWrapper< To, Tr, Tgemm >::set_nthreads().

◆ _Msize

unsigned int _Msize

Definition at line 143 of file arm_gemm.hpp.

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

◆ _nbatches

unsigned int _nbatches

Definition at line 147 of file arm_gemm.hpp.

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

◆ _nmulti

unsigned int _nmulti

Definition at line 148 of file arm_gemm.hpp.

◆ _Nsize

unsigned int _Nsize

Definition at line 144 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