Compute Library
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 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 

Definition at line 159 of file arm_gemm.hpp.

163  : _ci(ci), _Msize(M), _Nsize(N), _Ksize(K), _Ksections(Ksections), _nbatches(nbatches), _nmulti(nmulti), _indirect_input(indirect_input), _act(act), _maxthreads(maxthreads),
164  _fixed_format(fixed_format), _fast_mode(fast_mode), _cfg(cfg)
165  {
166  }
const CPUInfo * _ci
Definition: arm_gemm.hpp:145
unsigned int _nmulti
Definition: arm_gemm.hpp:151
unsigned int _Nsize
Definition: arm_gemm.hpp:147
Activation _act
Definition: arm_gemm.hpp:153
unsigned int M
const GemmConfig * _cfg
Definition: arm_gemm.hpp:157
unsigned int N
unsigned int _Msize
Definition: arm_gemm.hpp:146
unsigned int _Ksections
Definition: arm_gemm.hpp:149
unsigned int _Ksize
Definition: arm_gemm.hpp:148
unsigned int _nbatches
Definition: arm_gemm.hpp:150
unsigned int K

Field Documentation

◆ _act

◆ _cfg

◆ _ci

◆ _fast_mode

bool _fast_mode

Definition at line 156 of file arm_gemm.hpp.

◆ _fixed_format

◆ _indirect_input

◆ _Ksections

◆ _Ksize

◆ _maxthreads

◆ _Msize

◆ _nbatches

◆ _nmulti

◆ _Nsize

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