Compute Library
 23.08
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 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  }

Member Function Documentation

◆ get_working_size()

size_t get_working_size ( unsigned int  n_threads) const
inlinefinaloverride

Definition at line 292 of file depthfirst_driver.hpp.

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

The documentation for this class was generated from the following file:
GemmTuner.args
args
Definition: GemmTuner.py:679
strategy
const StratType * strategy
Definition: working_space.hpp:105