24.02.1
|
#include <sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp>
Public Types | |
using | return_type = __fp16 |
Public Types inherited from DepthwiseDepthfirstStrategy< __fp16, __fp16, __fp16, __fp16 > | |
typedef void(* | IndirectKernelType) (const __fp16 *const *input_ptrs, __fp16 *const *output_ptrs, const void *params, unsigned int n_channels, const __fp16 activation_min, const __fp16 activation_max) |
typedef void(* | DirectKernelType) (const unsigned int n_tile_rows, const unsigned int n_tile_cols, const __fp16 *inptr_base, int64_t ld_input_row, int64_t ld_input_col, __fp16 *outptr_base, int64_t ld_output_row, int64_t ld_output_col, const void *params, unsigned int n_channels, const __fp16 activation_min, const __fp16 activation_max) |
Public Member Functions | |
sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst (const CPUInfo *) | |
arm_gemm::VLType | get_vl_type (void) const override |
Parent::IndirectKernelType | get_indirect_kernel () const override |
Parent::DirectKernelType | get_direct_kernel () const override |
Public Member Functions inherited from DepthwiseDepthfirstStrategyCommon< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type > | |
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 DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type > | |
virtual size_t | get_storage_size (const DepthwiseArgs &args) const |
virtual void | pack_parameters (const DepthwiseArgs &args, void *buffer, const void *biases, const typename DefaultOutputStage< __fp16 >::Type &, const void *weights, size_t ld_weight_col, size_t ld_weight_row) const |
Public Member Functions inherited from DepthfirstStrategyUntyped | |
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 |
Static Public Attributes | |
constexpr static auto | vl_type = arm_gemm::VLType::SME |
constexpr static unsigned int | kernel_rows = 3 |
constexpr static unsigned int | kernel_cols = 3 |
constexpr static unsigned int | stride_rows = 1 |
constexpr static unsigned int | stride_cols = 1 |
constexpr static unsigned int | output_rows = 2 |
constexpr static unsigned int | output_cols = 2 |
Definition at line 37 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
using return_type = __fp16 |
Definition at line 45 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
|
inline |
Definition at line 57 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
|
inlineoverridevirtual |
Implements DepthwiseDepthfirstStrategy< __fp16, __fp16, __fp16, __fp16 >.
Definition at line 63 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
|
inlineoverridevirtual |
Implements DepthwiseDepthfirstStrategy< __fp16, __fp16, __fp16, __fp16 >.
Definition at line 62 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
|
inlineoverridevirtual |
Implements DepthfirstStrategyUntyped.
Definition at line 60 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
References sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst::vl_type.
|
staticconstexpr |
Definition at line 49 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
|
staticconstexpr |
Definition at line 48 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
|
staticconstexpr |
Definition at line 55 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
|
staticconstexpr |
Definition at line 54 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
|
staticconstexpr |
Definition at line 52 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
|
staticconstexpr |
Definition at line 51 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
|
staticconstexpr |
Definition at line 46 of file sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst.hpp.
Referenced by sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst::get_vl_type().