Compute Library
 22.08
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 349 of file depthwise_depthfirst_multiplier.hpp.

Member Typedef Documentation

◆ Type

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

Definition at line 351 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 354 of file depthwise_depthfirst_multiplier.hpp.

References 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.

359  {
360  strat->get_kernel()(
361  ws->input_rows, ws->outptr_array,
362  reinterpret_cast<const TWeight *>(parameters),
363  bias == nullptr ? nullptr : reinterpret_cast<const TAccum *>(bias) + start_output_channel,
364  strat->get_kernel_rows() * strat->get_kernel_cols(),
365  args.channel_multiplier,
366  ws->activation_min, ws->activation_max
367  );
368  }
std::unique_ptr< ParametersLibrary > parameters
Definition: Framework.cpp:46
const int32_t * bias

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