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

#include <depthwise_planar.hpp>

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

Public Member Functions

 DepthwisePlanar (StrategyType *const strat, const DepthwiseArgs &args, const OutputStage &os={})
 
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
 
size_t get_working_size (unsigned int n_threads, unsigned int) const override
 

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::DepthwisePlanar< TInput, TWeight, TOutput, TAccum, OutputStage >

Definition at line 239 of file depthwise_planar.hpp.

Constructor & Destructor Documentation

◆ DepthwisePlanar()

DepthwisePlanar ( StrategyType *const  strat,
const DepthwiseArgs &  args,
const OutputStage &  os = {} 
)
inline

Definition at line 254 of file depthwise_planar.hpp.

254  {})
255  : Parent(args), m_strat(strat), m_bias(nullptr), m_os(os)
256  {
257  }

Member Function Documentation

◆ get_storage_size()

size_t get_storage_size ( void  ) const
inlineoverride

Definition at line 259 of file depthwise_planar.hpp.

260  {
261  return m_strat->get_storage_size(this->m_args);
262  }

◆ get_working_size()

size_t get_working_size ( unsigned int  n_threads,
unsigned  int 
) const
inlineoverride

Definition at line 274 of file depthwise_planar.hpp.

References batches, bias, arm_compute::test::validation::input, input_height, input_width, and arm_compute::test::parameters.

275  {
276  return this->get_working_size_per_thread() * n_threads;
277  }

◆ 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 264 of file depthwise_planar.hpp.

268  {
269  m_strat->pack_parameters(this->m_args, buffer, biases, {}, weights, ld_weight_col, ld_weight_row);
270  this->m_bias = reinterpret_cast<const TAccum *>(biases);
271  depthwise_depthfirst::stash_bias(this->m_os, biases);
272  }

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