Compute Library
DepthfirstDriver< TInput, TWeight, TOutput > Class Template Referenceabstract

#include <depthfirst_driver.hpp>

Collaboration diagram for DepthfirstDriver< TInput, TWeight, TOutput >:

Public Member Functions

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

Definition at line 66 of file depthfirst_driver.hpp.

Constructor & Destructor Documentation

◆ DepthfirstDriver()

DepthfirstDriver ( IDepthfirstStrategy strategy,
const DepthwiseArgs &  args 

Definition at line 269 of file depthfirst_driver.hpp.

270  : Parent(args), m_strat(strategy)
271  {
272  }
const StratType * strategy

Member Function Documentation

◆ get_working_size()

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

Definition at line 274 of file depthfirst_driver.hpp.

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

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