PlanarStrategy< TInput, TWeight, TOutput, TAccum, OutputStage > Class Template Reference

#include <depthwise_planar.hpp>

Collaboration diagram for PlanarStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >:

Public Types
using	KernelType = typename PlanarKernelType< TInput, TWeight, TOutput, TAccum, OutputStage >::Type

Public Member Functions
	PlanarStrategy (unsigned int kernel_rows, unsigned int kernel_cols, unsigned int stride_rows, unsigned int stride_cols, unsigned int output_rows, arm_gemm::VLType vl_type)
unsigned int	get_output_rows (void) const override
arm_gemm::VLType	get_vl_type (void) const override
size_t	get_storage_size (const DepthwiseArgs &args) const override
void	pack_parameters (const DepthwiseArgs &args, void *buffer, const void *biases, const OutputStage &, const void *weights, size_t ld_weight_col, size_t ld_weight_row) const override
virtual KernelType	get_kernel (void) const =0
Public Member Functions inherited from IPlanarStrategy< typename DefaultOutputStage< TInput >::Type >
virtual	~IPlanarStrategy ()=default
virtual void	pack_parameters (const DepthwiseArgs &args, void *buffer, const void *biases, const typename DefaultOutputStage< TInput >::Type &, const void *weights, size_t ld_weight_col, size_t ld_weight_row) const=0

Detailed Description

template<typename TInput, typename TWeight = TInput, typename TOutput = TInput, typename TAccum = typename DefaultTAccum<TOutput>::Type, typename OutputStage = typename DefaultOutputStage<TOutput>::Type>
class arm_conv::depthwise::PlanarStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >

Definition at line 132 of file depthwise_planar.hpp.

Member Typedef Documentation

KernelType

using KernelType = typename PlanarKernelType<TInput, TWeight, TOutput, TAccum, OutputStage>::Type

Definition at line 195 of file depthwise_planar.hpp.

Constructor & Destructor Documentation

PlanarStrategy()

PlanarStrategy ( unsigned int  kernel_rows, unsigned int  kernel_cols, unsigned int  stride_rows, unsigned int  stride_cols, unsigned int  output_rows, arm_gemm::VLType  vl_type  )

inline

Definition at line 164 of file depthwise_planar.hpp.

    : m_kernel_rows(kernel_rows), m_kernel_cols(kernel_cols),
      m_stride_rows(stride_rows), m_stride_cols(stride_cols),
      m_output_rows(output_rows), m_vl_type(vl_type)
  {
  }

Member Function Documentation

get_kernel()

virtual KernelType get_kernel ( void  ) const

pure virtual

Implemented in sme2_fp32_planar_3x3_s1_4rows_mla_za, sme2_fp32_planar_3x3_s2_4rows_mla_za, sme2_fp32_planar_5x5_s1_4rows_mla_za, sme2_fp32_planar_5x5_s2_4rows_mla_za, sme2_fp32bf16fp32_planar_3x3_s1_4rows_dot_za, sme2_fp32bf16fp32_planar_3x3_s2_4rows_dot_za, sme2_fp32bf16fp32_planar_5x5_s1_4rows_dot_za, sme2_fp32bf16fp32_planar_5x5_s2_4rows_dot_za, sme2_s8q_planar_3x3_s1_4rows_dot_za, sme2_s8q_planar_3x3_s2_4rows_dot_za, sme2_s8q_planar_5x5_s1_4rows_dot_za, sme2_s8q_planar_5x5_s2_4rows_dot_za, sme2_u8q_planar_3x3_s1_4rows_dot_za, sme2_u8q_planar_3x3_s2_4rows_dot_za, sme2_u8q_planar_5x5_s1_4rows_dot_za, sme2_u8q_planar_5x5_s2_4rows_dot_za, sme2_u8s8u8q_planar_3x3_s1_4rows_dot_za, sme2_u8s8u8q_planar_3x3_s2_4rows_dot_za, sme2_u8s8u8q_planar_5x5_s1_4rows_dot_za, and sme2_u8s8u8q_planar_5x5_s2_4rows_dot_za.

get_output_rows()

unsigned int get_output_rows ( void  ) const

inlineoverridevirtual

Implements IPlanarStrategy< typename DefaultOutputStage< TInput >::Type >.

Definition at line 175 of file depthwise_planar.hpp.

{ return m_output_rows; }

get_storage_size()

size_t get_storage_size ( const DepthwiseArgs &  args ) const

inlineoverridevirtual

Implements IPlanarStrategy< typename DefaultOutputStage< TInput >::Type >.

Definition at line 178 of file depthwise_planar.hpp.

  {
    return interleaves::get_storage_size_generic(this->get_kernel_packing_arguments(), args);
  }

get_vl_type()

arm_gemm::VLType get_vl_type ( void  ) const

inlineoverridevirtual

Implements IPlanarStrategy< typename DefaultOutputStage< TInput >::Type >.

Definition at line 176 of file depthwise_planar.hpp.

{ return m_vl_type; }

pack_parameters()

void pack_parameters ( const DepthwiseArgs &  args, void *  buffer, const void *  biases, const OutputStage &  , const void *  weights, size_t  ld_weight_col, size_t  ld_weight_row  ) const

inlineoverride

Definition at line 183 of file depthwise_planar.hpp.

  {
    interleaves::pack_parameters_generic(
      this->get_kernel_packing_arguments(), args,
      buffer, biases, weights, ld_weight_col, ld_weight_row
    );
  }

---

The documentation for this class was generated from the following file:

• src/core/NEON/kernels/arm_conv/depthwise/depthwise_planar.hpp