24.02.1
|
#include <gemv_batched.hpp>
Public Member Functions | |
GemvBatched (const GemmArgs &args) | |
void | set_arrays (const To *A, const int, const int A_batch_stride, const int A_multi_stride, const To *B, const int ldb, const int B_multi_stride, Tr *C, const int, const int C_batch_stride, const int C_multi_stride, const Tr *bias, const int bias_multi_stride) override |
ndrange_t | get_window_size () const override |
void | set_nthreads (int nthreads) override |
void | execute (const ndcoord_t &work_range, const ndcoord_t &thread_locator, int threadid) override |
Main execute member fucntion. More... | |
size_t | get_working_size () const override |
void | set_working_space (void *space) override |
bool | B_is_pretransposed () const override |
bool | B_pretranspose_required () const override |
size_t | get_B_pretransposed_array_size () const override |
void | pretranspose_B_array (void *buffer, const To *B, const int ldb, const int B_multi_stride) override |
void | set_pretransposed_B_data (void *buffer) override |
GemmConfig | get_config () override |
Public Member Functions inherited from GemmCommon< To, Tr > | |
void | set_arrays_generic (const void *A, const int lda, const int A_batch_stride, const int A_multi_stride, const void *B, const int ldb, const int B_multi_stride, void *C, const int ldc, const int C_batch_stride, const int C_multi_stride, const void *bias, const int bias_multi_stride) override |
virtual void | requantize_bias (void *, const To *, const int, const int) |
void | pretranspose_B_array_generic (void *out, const void *in, const int row_stride, const int multi_stride) override |
virtual void | pretranspose_B_array_part (void *out, const To *in, const int row_stride, const int multi_stride, size_t, size_t) |
void | pretranspose_B_array_part_generic (void *out, const void *in, const int row_stride, const int multi_stride, size_t start, size_t end) override |
virtual void | set_indirect_parameters (size_t, const To *const *const *) |
void | set_indirect_parameters_generic (size_t sz, const void *const *const *ptr) override |
Public Member Functions inherited from IGemmCommon | |
virtual bool | supports_dynamic_scheduling () const |
virtual size_t | get_B_pretranspose_window_size () const |
virtual void | set_quantized_bias (const int32_t *, size_t) |
virtual void | set_convolution_parameters (ConvolutionParameters) |
virtual | ~IGemmCommon () |
Definition at line 34 of file gemv_batched.hpp.
|
inline |
Definition at line 39 of file gemv_batched.hpp.
References GemmArgs::_cfg, GemmArgs::_Msize, GemmArgs::_nbatches, and GemmTuner::args.
|
inlineoverridevirtual |
Reimplemented from IGemmCommon.
Definition at line 79 of file gemv_batched.hpp.
|
inlineoverridevirtual |
Reimplemented from IGemmCommon.
Definition at line 83 of file gemv_batched.hpp.
|
inlineoverridevirtual |
Main execute member fucntion.
[in] | work_range | specifies the range of work we want to be computed, total range defined by get_window_size() |
[in] | thread_locator | where are we inside of the thread space |
[in] | threadid | a unique threadid |
Implements IGemmCommon.
Definition at line 67 of file gemv_batched.hpp.
|
inlineoverridevirtual |
Reimplemented from IGemmCommon.
Definition at line 87 of file gemv_batched.hpp.
|
inlineoverridevirtual |
Implements IGemmCommon.
Definition at line 99 of file gemv_batched.hpp.
References GemmConfig::filter.
|
inlineoverridevirtual |
Implements IGemmCommon.
Definition at line 59 of file gemv_batched.hpp.
|
inlineoverridevirtual |
Reimplemented from IGemmCommon.
Definition at line 71 of file gemv_batched.hpp.
|
inlineoverridevirtual |
Reimplemented from GemmCommon< To, Tr >.
Definition at line 91 of file gemv_batched.hpp.
|
inlineoverridevirtual |
Reimplemented from GemmCommon< To, Tr >.
Definition at line 48 of file gemv_batched.hpp.
References bias.
|
inlineoverridevirtual |
Reimplemented from IGemmCommon.
Definition at line 63 of file gemv_batched.hpp.
|
inlineoverridevirtual |
Reimplemented from IGemmCommon.
Definition at line 95 of file gemv_batched.hpp.
|
inlineoverridevirtual |
Reimplemented from IGemmCommon.
Definition at line 75 of file gemv_batched.hpp.