24.04
|
Go to the documentation of this file.
35 void sme2_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst_direct_impl(
const unsigned int n_tile_rows,
const unsigned int n_tile_cols,
const __fp16 *inptr, int64_t ld_input_row, int64_t ld_input_col, __fp16 *outptr, 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);
template UniqueDepthwiseCommon< float > depthwise(const DepthwiseArgs &, const Nothing &)
constexpr static unsigned int kernel_rows
constexpr static unsigned int stride_rows
constexpr static unsigned int output_rows
constexpr static unsigned int output_cols
sme2_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst(const CPUInfo *)
void sme2_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst_indirect_impl(const __fp16 *const *const input_ptrs, __fp16 *const *const outptrs, const void *params, unsigned int n_channels, const __fp16 activation_min, const __fp16 activation_max)
arm_gemm::VLType get_vl_type(void) const override
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)
Parent::DirectKernelType get_direct_kernel() const override
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)
constexpr static auto vl_type
void sme2_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst_direct_impl(const unsigned int n_tile_rows, const unsigned int n_tile_cols, const __fp16 *inptr, int64_t ld_input_row, int64_t ld_input_col, __fp16 *outptr, 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)
constexpr static unsigned int stride_cols
Parent::IndirectKernelType get_indirect_kernel() const override
constexpr static unsigned int kernel_cols