23.11
|
#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 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 |
Definition at line 139 of file arm_gemm.hpp.
|
inline |
Definition at line 156 of file arm_gemm.hpp.
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().
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().
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().
bool _fast_mode |
Definition at line 153 of file arm_gemm.hpp.
bool _fixed_format |
Definition at line 152 of file arm_gemm.hpp.
bool _indirect_input |
Definition at line 149 of file arm_gemm.hpp.
Referenced by GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::execute().
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().
unsigned int _Ksize |
Definition at line 145 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(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::pretranspose_B_array_part(), GemvPretransposed< strategy, To, Tr, OutputStage >::requantize_bias(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::requantize_bias(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::set_convolution_parameters(), and GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::set_indirect_parameters().
int _maxthreads |
Definition at line 151 of file arm_gemm.hpp.
Referenced by QuantizeWrapper< To, Tr, Tgemm >::set_nthreads().
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().
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().
unsigned int _nmulti |
Definition at line 148 of file arm_gemm.hpp.
Referenced by GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::get_B_pretranspose_window_size(), GemvPretransposed< strategy, To, Tr, OutputStage >::get_B_pretransposed_array_size(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::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, FixedFormat >::pretranspose_B_array_part(), GemvPretransposed< strategy, To, Tr, OutputStage >::requantize_bias(), and GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::requantize_bias().
unsigned int _Nsize |
Definition at line 144 of file arm_gemm.hpp.
Referenced by GemvPretransposed< strategy, To, Tr, OutputStage >::execute(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::execute(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::get_B_pretranspose_window_size(), GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::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, FixedFormat >::pretranspose_B_array_part(), GemvPretransposed< strategy, To, Tr, OutputStage >::requantize_bias(), and GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::requantize_bias().