Compute Library
 23.08
GenericDepthfirstMultiplierKernelStrategy< TInput, TWeight, TOutput, TAccum > Class Template Referenceabstract

#include <depthwise_depthfirst_multiplier.hpp>

Public Types

using KernelType = std::function< void(const TInput *const *, TOutput *const *, const TWeight *, const TAccum *, unsigned int, unsigned int, TAccum, TAccum)>
 

Public Member Functions

 GenericDepthfirstMultiplierKernelStrategy (unsigned int output_rows, unsigned int output_cols, arm_gemm::VLType vl_type)
 
virtual ~GenericDepthfirstMultiplierKernelStrategy ()=default
 
arm_gemm::VLType get_vl_type (void) const
 
unsigned int get_output_rows (void) const
 
unsigned int get_output_cols (void) const
 
virtual KernelType get_kernel (void) const =0
 

Detailed Description

template<typename TInput, typename TWeight, typename TOutput, typename TAccum>
class arm_conv::depthwise::GenericDepthfirstMultiplierKernelStrategy< TInput, TWeight, TOutput, TAccum >

Definition at line 126 of file depthwise_depthfirst_multiplier.hpp.

Member Typedef Documentation

◆ KernelType

using KernelType = std::function<void( const TInput *const *, TOutput *const *, const TWeight *, const TAccum *, unsigned int, unsigned int, TAccum, TAccum )>

Definition at line 150 of file depthwise_depthfirst_multiplier.hpp.

Constructor & Destructor Documentation

◆ GenericDepthfirstMultiplierKernelStrategy()

GenericDepthfirstMultiplierKernelStrategy ( unsigned int  output_rows,
unsigned int  output_cols,
arm_gemm::VLType  vl_type 
)
inline

Definition at line 132 of file depthwise_depthfirst_multiplier.hpp.

133  : m_vl_type(vl_type), m_output_rows(output_rows), m_output_cols(output_cols)
134  {
135  }

◆ ~GenericDepthfirstMultiplierKernelStrategy()

virtual ~GenericDepthfirstMultiplierKernelStrategy ( )
virtualdefault

Member Function Documentation

◆ get_kernel()

virtual KernelType get_kernel ( void  ) const
pure virtual

◆ get_output_cols()

unsigned int get_output_cols ( void  ) const
inline

Definition at line 141 of file depthwise_depthfirst_multiplier.hpp.

141 { return m_output_cols; }

◆ get_output_rows()

unsigned int get_output_rows ( void  ) const
inline

Definition at line 140 of file depthwise_depthfirst_multiplier.hpp.

140 { return m_output_rows; }

◆ get_vl_type()

arm_gemm::VLType get_vl_type ( void  ) const
inline

Definition at line 139 of file depthwise_depthfirst_multiplier.hpp.

139 { return m_vl_type; }

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