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

#include <depthwise_planar.hpp>

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

Public Member Functions
	DepthwisePlanar (StrategyType *const strat, const DepthwiseArgs &args, const OutputStage &os={})
	DepthwisePlanar (DepthwisePlanar &)=delete
DepthwisePlanar &	operator= (DepthwisePlanar &)=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
size_t	get_working_size (unsigned int n_threads) 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 241 of file depthwise_planar.hpp.

Constructor & Destructor Documentation

DepthwisePlanar() [1/2]

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

inline

Definition at line 256 of file depthwise_planar.hpp.

                                                                                                {})
  : Parent(args), m_strat(strat), m_bias(nullptr), m_os(os)
  {
  }

DepthwisePlanar() [2/2]

DepthwisePlanar ( DepthwisePlanar< TInput, TWeight, TOutput, TAccum, OutputStage > &  )

delete

Member Function Documentation

get_storage_size()

size_t get_storage_size ( void  ) const

inlineoverride

Definition at line 264 of file depthwise_planar.hpp.

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

get_working_size()

size_t get_working_size ( unsigned int  n_threads ) const

inlineoverride

Definition at line 279 of file depthwise_planar.hpp.

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

operator=()

DepthwisePlanar& operator= ( DepthwisePlanar< 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 269 of file depthwise_planar.hpp.

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

---

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

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