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

#include <depthwise_depthfirst_multiplier.hpp>

Public Types

using KernelType = std::function< void(const TInput *const *, TOutput *const *, const TWeight *, const int32_t *, unsigned int, unsigned int, const int32_t *, const int32_t *, const int32_t *, const arm_gemm::Requantize32 &)>
 

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>
class arm_conv::depthwise::GenericDepthfirstMultiplierKernelStrategy< TInput, TWeight, TOutput, int32_t >

Definition at line 155 of file depthwise_depthfirst_multiplier.hpp.

Member Typedef Documentation

◆ KernelType

using KernelType = std::function<void( const TInput *const *, TOutput *const *, const TWeight *, const int32_t *, unsigned int, unsigned int, const int32_t *, const int32_t *, const int32_t *, const arm_gemm::Requantize32 & )>

Definition at line 180 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 161 of file depthwise_depthfirst_multiplier.hpp.

162  : m_vl_type(vl_type), m_output_rows(output_rows), m_output_cols(output_cols)
163  {
164  }

◆ ~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 170 of file depthwise_depthfirst_multiplier.hpp.

170 { return m_output_cols; }

◆ get_output_rows()

unsigned int get_output_rows ( void  ) const
inline

Definition at line 169 of file depthwise_depthfirst_multiplier.hpp.

169 { return m_output_rows; }

◆ get_vl_type()

arm_gemm::VLType get_vl_type ( void  ) const
inline

Definition at line 168 of file depthwise_depthfirst_multiplier.hpp.

168 { return m_vl_type; }

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