DepthfirstDriver< TInput, TWeight, TOutput > Class Template Reference

#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) const override final
virtual bool	supports_direct_padding () const

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  )

inline

Definition at line 273 of file depthfirst_driver.hpp.

  : Parent(args), m_strat(strategy)
  {
  }

Member Function Documentation

get_working_size()

size_t get_working_size ( unsigned int  n_threads ) const

inlinefinaloverride

Definition at line 278 of file depthfirst_driver.hpp.

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

supports_direct_padding()

virtual bool supports_direct_padding ( ) const

inlinevirtual

Definition at line 283 of file depthfirst_driver.hpp.

  {
    return false;
  }

---

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

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