22.05
|
#include <arm_gemm.hpp>
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 |
Definition at line 101 of file arm_gemm.hpp.
|
inline |
Definition at line 117 of file arm_gemm.hpp.
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().
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().
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().
bool _fast_mode |
Definition at line 114 of file arm_gemm.hpp.
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().
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().
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().
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().
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().
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().
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().
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().