24.02.1
|
#include <depthwise_depthfirst.hpp>
Public Types | |
typedef void(* | KernelType) (unsigned int, const TInput *const *, const TWeight *, const int32_t *, const arm_gemm::Requantize32 &, const int32_t *, const int32_t *, TOutput *const *) |
Public Member Functions | |
virtual KernelType | get_kernel () const =0 |
size_t | get_storage_size (const DepthwiseArgs &args) const override |
void | pack_parameters (const DepthwiseArgs &args, void *buffer, const void *biases, const arm_gemm::Requantize32 &, const void *weights, size_t ld_weight_col, size_t ld_weight_row) const override |
Public Member Functions inherited from DepthwiseDepthfirstStrategyCommon< TInput, TWeight, TOutput, int32_t, arm_gemm::Requantize32 > | |
DepthwiseDepthfirstStrategyCommon (unsigned int output_rows, unsigned int output_cols, unsigned int kernel_rows, unsigned int kernel_cols, unsigned int stride_rows=1, unsigned int stride_cols=1) | |
DepthwiseDepthfirstStrategyCommon (unsigned int output_size, unsigned int kernel_size, unsigned int stride=1) | |
virtual | ~DepthwiseDepthfirstStrategyCommon () |
unsigned int | get_output_rows () const override |
unsigned int | get_output_cols () const override |
unsigned int | get_kernel_rows () const override |
unsigned int | get_kernel_cols () const override |
unsigned int | get_stride_rows () const override |
unsigned int | get_stride_cols () const override |
Public Member Functions inherited from DepthfirstStrategyUntyped | |
virtual arm_gemm::VLType | get_vl_type () const =0 |
virtual unsigned int | get_kernel_rows () const =0 |
virtual unsigned int | get_kernel_cols () const =0 |
virtual unsigned int | get_stride_rows () const =0 |
virtual unsigned int | get_stride_cols () const =0 |
virtual unsigned int | get_input_rows () const override |
virtual unsigned int | get_input_cols () const override |
virtual unsigned int | get_n_input_points () const |
virtual unsigned int | get_n_output_points () const |
virtual unsigned int | get_n_kernel_points () const |
virtual bool | uses_premultiply () const |
virtual unsigned int | get_accumulator_depth_vl () const |
virtual bool | get_kernel_packing_point (const unsigned int index, unsigned int &x, unsigned int &y) const |
Public Member Functions inherited from IDepthfirstStrategy | |
virtual | ~IDepthfirstStrategy ()=default |
virtual unsigned int | get_output_rows () const =0 |
virtual unsigned int | get_output_cols () const =0 |
Definition at line 108 of file depthwise_depthfirst.hpp.
typedef void(* KernelType(unsigned int, const TInput *const *, const TWeight *, const int32_t *, const arm_gemm::Requantize32 &, const int32_t *, const int32_t *, TOutput *const *) |
Definition at line 128 of file depthwise_depthfirst.hpp.
|
pure virtual |
|
inlineoverridevirtual |
Reimplemented from DepthfirstStrategy< TInput, TWeight, TOutput, int32_t, arm_gemm::Requantize32 >.
Definition at line 139 of file depthwise_depthfirst.hpp.
References GemmTuner::args, and arm_conv::depthwise::interleaves::get_storage_size_generic().
|
inlineoverridevirtual |
Reimplemented from DepthfirstStrategy< TInput, TWeight, TOutput, int32_t, arm_gemm::Requantize32 >.
Definition at line 144 of file depthwise_depthfirst.hpp.
References GemmTuner::args, and arm_conv::depthwise::interleaves::pack_parameters_generic().