Compute Library
 23.11
GemmArgs Struct Reference

#include <arm_gemm.hpp>

Collaboration diagram for GemmArgs:
[legend]

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.

169  : _ci(ci),
170  _Msize(M),
171  _Nsize(N),
172  _Ksize(K),
173  _Ksections(Ksections),
174  _nbatches(nbatches),
175  _nmulti(nmulti),
176  _indirect_input(indirect_input),
177  _act(act),
178  _maxthreads(maxthreads),
179  _fixed_format(fixed_format),
180  _fast_mode(fast_mode),
181  _cfg(cfg)
182  {
183  }

Field Documentation

◆ _act

◆ _cfg

◆ _ci

◆ _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

◆ _Ksections

◆ _Ksize

◆ _maxthreads

int _maxthreads

Definition at line 151 of file arm_gemm.hpp.

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

◆ _Msize

◆ _nbatches

◆ _nmulti

◆ _Nsize


The documentation for this struct was generated from the following file:
N
unsigned int N
Definition: CpuGemmAssemblyDispatch.cpp:103
arm_gemm::GemmArgs::_ci
const CPUInfo * _ci
Definition: arm_gemm.hpp:142
arm_gemm::GemmArgs::_nbatches
unsigned int _nbatches
Definition: arm_gemm.hpp:147
arm_gemm::GemmArgs::_indirect_input
bool _indirect_input
Definition: arm_gemm.hpp:149
arm_gemm::GemmArgs::_cfg
const GemmConfig * _cfg
Definition: arm_gemm.hpp:154
arm_gemm::GemmArgs::_Msize
unsigned int _Msize
Definition: arm_gemm.hpp:143
arm_gemm::GemmArgs::_fixed_format
bool _fixed_format
Definition: arm_gemm.hpp:152
arm_gemm::GemmArgs::_Ksections
unsigned int _Ksections
Definition: arm_gemm.hpp:146
arm_gemm::GemmArgs::_fast_mode
bool _fast_mode
Definition: arm_gemm.hpp:153
arm_gemm::GemmArgs::_Nsize
unsigned int _Nsize
Definition: arm_gemm.hpp:144
K
unsigned int K
Definition: CpuGemmAssemblyDispatch.cpp:104
ci
const CPUInfo & ci
Definition: NEBatchNormalizationLayerKernel.cpp:51
arm_gemm::GemmArgs::_maxthreads
int _maxthreads
Definition: arm_gemm.hpp:151
M
unsigned int M
Definition: CpuGemmAssemblyDispatch.cpp:102
arm_gemm::GemmArgs::_act
Activation _act
Definition: arm_gemm.hpp:150
arm_gemm::GemmArgs::_Ksize
unsigned int _Ksize
Definition: arm_gemm.hpp:145
arm_gemm::GemmArgs::_nmulti
unsigned int _nmulti
Definition: arm_gemm.hpp:148