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

#include <depthfirst_driver.hpp>

Collaboration diagram for DepthfirstDriver< TInput, TOutput >:
[legend]

Public Member Functions

 DepthfirstDriver (const IDepthfirstStrategy *strategy, const PoolingArgs &args)
 
size_t get_working_size (unsigned int n_threads) const override
 
size_t get_working_size (unsigned int n_threads, unsigned int n_channels) const override final
 

Detailed Description

template<typename TInput, typename TOutput>
class arm_conv::pooling::DepthfirstDriver< TInput, TOutput >

Definition at line 58 of file depthfirst_driver.hpp.

Constructor & Destructor Documentation

◆ DepthfirstDriver()

DepthfirstDriver ( const IDepthfirstStrategy strategy,
const PoolingArgs &  args 
)
inline

Definition at line 287 of file depthfirst_driver.hpp.

288  : Parent(args), m_strat(strategy)
289  {
290  }
const StratType * strategy

Member Function Documentation

◆ get_working_size() [1/2]

size_t get_working_size ( unsigned int  n_threads) const
inlineoverride

Definition at line 292 of file depthfirst_driver.hpp.

293  {
294  return this->get_working_size(n_threads, this->m_args.n_channels);
295  }
size_t get_working_size(unsigned int n_threads) const override

◆ get_working_size() [2/2]

size_t get_working_size ( unsigned int  n_threads,
unsigned int  n_channels 
) const
inlinefinaloverride

Definition at line 297 of file depthfirst_driver.hpp.

298  {
299  return n_threads * this->get_working_size_per_thread(n_channels);
300  }

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