StrategyType< true, TInput, TWeight, TOutput, TAccum, OutputStage > Struct Template Reference

#include <depthwise_depthfirst_multiplier.hpp>

Public Types
using	Type = GenericDepthfirstMultiplierStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >

Static Public Member Functions
template<typename WorkspaceType >
static void	execute (const DepthwiseArgs &args, const WorkspaceType *ws, const Type *strat, const OutputStage &, const unsigned int start_output_channel, const void *parameters, const void *bias)

Detailed Description

template<typename TInput, typename TWeight, typename TOutput, typename TAccum, typename OutputStage>
struct arm_conv::depthwise::depthfirst_multiplier::StrategyType< true, TInput, TWeight, TOutput, TAccum, OutputStage >

Definition at line 353 of file depthwise_depthfirst_multiplier.hpp.

Member Typedef Documentation

Type

using Type = GenericDepthfirstMultiplierStrategy<TInput, TWeight, TOutput, TAccum, OutputStage>

Definition at line 355 of file depthwise_depthfirst_multiplier.hpp.

Member Function Documentation

execute()

static void execute ( const DepthwiseArgs &  args, const WorkspaceType *  ws, const Type *  strat, const OutputStage &  , const unsigned int  start_output_channel, const void *  parameters, const void *  bias  )

inlinestatic

Definition at line 358 of file depthwise_depthfirst_multiplier.hpp.

  {
    strat->get_kernel()(
      ws->input_rows, ws->outptr_array,
      reinterpret_cast<const TWeight *>(parameters),
      bias == nullptr ? nullptr : reinterpret_cast<const TAccum *>(bias) + start_output_channel,
      strat->get_kernel_rows() * strat->get_kernel_cols(),
      args.channel_multiplier,
      ws->activation_min, ws->activation_max
    );
  }

References GemmTuner::args, bias, GenericDepthfirstMultiplierStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::get_kernel(), DepthwiseDepthfirstStrategyCommon< TInput, TWeight, TOutput, TAccum, OutputStage >::get_kernel_cols(), DepthwiseDepthfirstStrategyCommon< TInput, TWeight, TOutput, TAccum, OutputStage >::get_kernel_rows(), and arm_compute::test::parameters.

---

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

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