GemmArgs Struct Reference

#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 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	_fast_mode
const GemmConfig *	_cfg

Detailed Description

Definition at line 101 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  fast_mode = false, const GemmConfig *  cfg = nullptr  )

inline

Definition at line 117 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), _fast_mode(fast_mode),
          _cfg(cfg)
    {
    }

Field Documentation

_act

Activation _act

Definition at line 112 of file arm_gemm.hpp.

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

_cfg

const GemmConfig* _cfg

Definition at line 115 of file arm_gemm.hpp.

Referenced by arm_gemm::find_implementation(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::GemmHybridIndirect(), GemmInterleavedPretransposed2d< strategy, To, Tr >::GemmInterleavedPretransposed2d(), GemvBatched< To, Tr >::GemvBatched(), and GemvPretransposed< strategy, To, Tr, OutputStage >::GemvPretransposed().

_ci

const CPUInfo* _ci

Definition at line 104 of file arm_gemm.hpp.

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

_fast_mode

bool _fast_mode

Definition at line 114 of file arm_gemm.hpp.

_indirect_input

bool _indirect_input

Definition at line 111 of file arm_gemm.hpp.

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

_Ksections

unsigned int _Ksections

Definition at line 108 of file arm_gemm.hpp.

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

_Ksize

unsigned int _Ksize

Definition at line 107 of file arm_gemm.hpp.

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

_maxthreads

int _maxthreads

Definition at line 113 of file arm_gemm.hpp.

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

_Msize

unsigned int _Msize

Definition at line 105 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 >::estimate_cycles(), GemmInterleaved< strategy, To, Tr, OutputStage, MergeStep, ForceThreadColumns >::estimate_cycles(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::execute(), GemvBatched< To, Tr >::GemvBatched(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::get_B_pretransposed_array_size(), and QuantizeWrapper< To, Tr, Tgemm >::QuantizeWrapper().

_nbatches

unsigned int _nbatches

Definition at line 109 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 >::estimate_cycles(), GemmInterleaved< strategy, To, Tr, OutputStage, MergeStep, ForceThreadColumns >::estimate_cycles(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::execute(), GemvBatched< To, Tr >::GemvBatched(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::get_B_pretransposed_array_size(), and QuantizeWrapper< To, Tr, Tgemm >::QuantizeWrapper().

_nmulti

unsigned int _nmulti

Definition at line 110 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 >::estimate_cycles(), GemmInterleaved< strategy, To, Tr, OutputStage, MergeStep, ForceThreadColumns >::estimate_cycles(), GemvPretransposed< strategy, To, Tr, OutputStage >::get_B_pretransposed_array_size(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::get_B_pretransposed_array_size(), GemvPretransposed< strategy, To, Tr, OutputStage >::get_window_size(), GemvPretransposed< strategy, To, Tr, OutputStage >::pretranspose_B_array(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::pretranspose_B_array(), QuantizeWrapper< To, Tr, Tgemm >::QuantizeWrapper(), GemvPretransposed< strategy, To, Tr, OutputStage >::requantize_bias(), and GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::requantize_bias().

_Nsize

unsigned int _Nsize

Definition at line 106 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 >::estimate_cycles(), GemmInterleaved< strategy, To, Tr, OutputStage, MergeStep, ForceThreadColumns >::estimate_cycles(), GemvPretransposed< strategy, To, Tr, OutputStage >::execute(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::execute(), GemvPretransposed< strategy, To, Tr, OutputStage >::GemvPretransposed(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::get_B_pretransposed_array_size(), GemvPretransposed< strategy, To, Tr, OutputStage >::get_window_size(), GemvPretransposed< strategy, To, Tr, OutputStage >::pretranspose_B_array(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::pretranspose_B_array(), QuantizeWrapper< To, Tr, Tgemm >::QuantizeWrapper(), GemvPretransposed< strategy, To, Tr, OutputStage >::requantize_bias(), and GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::requantize_bias().

---

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

• src/cpu/kernels/assembly/arm_gemm.hpp