Compute Library
 22.08
DepthwiseDepthfirstCommon< TInput, TWeight, TOutput, TAccum, OutputStage > Class Template Reference

#include <depthwise_depthfirst.hpp>

Collaboration diagram for DepthwiseDepthfirstCommon< TInput, TWeight, TOutput, TAccum, OutputStage >:
[legend]

Public Member Functions

 DepthwiseDepthfirstCommon (StratType *const strat, const DepthwiseArgs &args, const OutputStage &os)
 
 DepthwiseDepthfirstCommon (DepthwiseDepthfirstCommon &)=delete
 
DepthwiseDepthfirstCommonoperator= (DepthwiseDepthfirstCommon &)=delete
 
size_t get_storage_size (void) const override
 
void pack_parameters (void *buffer, const void *biases, const void *weights, size_t ld_weight_col, size_t ld_weight_row) override
 
- Public Member Functions inherited from DepthfirstDriver< TInput, TWeight, TOutput >
 DepthfirstDriver (IDepthfirstStrategy *strategy, const DepthwiseArgs &args)
 
size_t get_working_size (unsigned int n_threads, unsigned int n_input_channels) const override final
 

Detailed Description

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

Definition at line 156 of file depthwise_depthfirst.hpp.

Constructor & Destructor Documentation

◆ DepthwiseDepthfirstCommon() [1/2]

DepthwiseDepthfirstCommon ( StratType *const  strat,
const DepthwiseArgs &  args,
const OutputStage &  os 
)
inline

Definition at line 166 of file depthwise_depthfirst.hpp.

167  : DepthfirstDriver<TInput, TWeight, TOutput>(strat, args), m_os(os)
168  {
169  }

◆ DepthwiseDepthfirstCommon() [2/2]

DepthwiseDepthfirstCommon ( DepthwiseDepthfirstCommon< TInput, TWeight, TOutput, TAccum, OutputStage > &  )
delete

Member Function Documentation

◆ get_storage_size()

size_t get_storage_size ( void  ) const
inlineoverride

Definition at line 174 of file depthwise_depthfirst.hpp.

References DepthfirstStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::get_storage_size().

175  {
176  return reinterpret_cast<const StratType *>(this->m_strat.get())->
177  get_storage_size(this->m_args);
178  }

◆ operator=()

DepthwiseDepthfirstCommon& operator= ( DepthwiseDepthfirstCommon< TInput, TWeight, TOutput, TAccum, OutputStage > &  )
delete

◆ pack_parameters()

void pack_parameters ( void *  buffer,
const void *  biases,
const void *  weights,
size_t  ld_weight_col,
size_t  ld_weight_row 
)
inlineoverride

Definition at line 180 of file depthwise_depthfirst.hpp.

References DepthfirstStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::pack_parameters().

181  {
182  reinterpret_cast<const StratType *>(this->m_strat.get())->
183  pack_parameters(this->m_args, buffer, biases, m_os, weights, ld_weight_col, ld_weight_row);
184  }
void pack_parameters(void *buffer, const void *biases, const void *weights, size_t ld_weight_col, size_t ld_weight_row) override

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