Compute Library
 22.11
PoolingImplementation< TInput, TOutput, OutputStage > Struct Template Reference

#include <pooling_implementation.hpp>

Public Member Functions

bool get_is_supported (const PoolingArgs &args, const OutputStage &os) const
 
uint64_t get_cycle_estimate (const PoolingArgs &args, const OutputStage &os) const
 
PoolingCommon< TInput, TOutput > * get_instance (const PoolingArgs &args, const OutputStage &os) const
 

Data Fields

const PoolingMethod method
 
const char * name
 
std::function< bool(const PoolingArgs &, const OutputStage &)> is_supported
 
std::function< uint64_t(const PoolingArgs &, const OutputStage &)> cycle_estimate
 
std::function< PoolingCommon< TInput, TOutput > *(const PoolingArgs &, const OutputStage &)> initialise
 

Detailed Description

template<typename TInput, typename TOutput, class OutputStage = Nothing>
struct arm_conv::pooling::PoolingImplementation< TInput, TOutput, OutputStage >

Definition at line 36 of file pooling_implementation.hpp.

Member Function Documentation

◆ get_cycle_estimate()

uint64_t get_cycle_estimate ( const PoolingArgs &  args,
const OutputStage &  os 
) const
inline

Definition at line 49 of file pooling_implementation.hpp.

References PoolingImplementation< TInput, TOutput, OutputStage >::cycle_estimate.

50  {
51  return (cycle_estimate == nullptr) ? 0 : cycle_estimate(args, os);
52  }
std::function< uint64_t(const PoolingArgs &, const OutputStage &)> cycle_estimate

◆ get_instance()

PoolingCommon<TInput, TOutput>* get_instance ( const PoolingArgs &  args,
const OutputStage &  os 
) const
inline

Definition at line 54 of file pooling_implementation.hpp.

References PoolingImplementation< TInput, TOutput, OutputStage >::initialise, and arm_conv::pooling::pooling_implementation_list().

Referenced by arm_conv::pooling::pooling().

55  {
56  return initialise(args, os);
57  }
std::function< PoolingCommon< TInput, TOutput > *(const PoolingArgs &, const OutputStage &)> initialise

◆ get_is_supported()

bool get_is_supported ( const PoolingArgs &  args,
const OutputStage &  os 
) const
inline

Definition at line 44 of file pooling_implementation.hpp.

References PoolingImplementation< TInput, TOutput, OutputStage >::is_supported.

45  {
46  return (is_supported == nullptr) ? true : is_supported(args, os);
47  }
std::function< bool(const PoolingArgs &, const OutputStage &)> is_supported

Field Documentation

◆ cycle_estimate

std::function<uint64_t(const PoolingArgs &, const OutputStage &)> cycle_estimate

◆ initialise

std::function<PoolingCommon<TInput, TOutput> *(const PoolingArgs &, const OutputStage &)> initialise

◆ is_supported

std::function<bool(const PoolingArgs &, const OutputStage &)> is_supported

◆ method

const PoolingMethod method

Definition at line 38 of file pooling_implementation.hpp.

◆ name

const char* name

Definition at line 39 of file pooling_implementation.hpp.


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