#include <depthwise_depthfirst.hpp>
|
template<typename Strat , typename Workspace > |
static void | indirect (const Strat *strat, const Workspace *ws, const OutputStage &, const void *params, const TAccum *, unsigned int n_channels) |
|
template<typename Strat , typename Workspace > |
static void | direct (const Strat *strat, const Workspace *ws, const OutputStage &, unsigned int n_tile_rows, unsigned int n_tile_cols, const TInput *inptr, size_t ld_in_row, size_t ld_in_col, TOutput *outptr, size_t ld_out_row, size_t ld_out_col, const void *params, unsigned int n_channels) |
|
template<typename TInput, typename TWeight, typename TOutput, typename TAccum, typename OutputStage>
struct arm_conv::depthwise::depthwise_depthfirst::Invoke< TInput, TWeight, TOutput, TAccum, OutputStage >
Definition at line 286 of file depthwise_depthfirst.hpp.
◆ direct()
static void direct |
( |
const Strat * |
strat, |
|
|
const Workspace * |
ws, |
|
|
const OutputStage & |
, |
|
|
unsigned int |
n_tile_rows, |
|
|
unsigned int |
n_tile_cols, |
|
|
const TInput * |
inptr, |
|
|
size_t |
ld_in_row, |
|
|
size_t |
ld_in_col, |
|
|
TOutput * |
outptr, |
|
|
size_t |
ld_out_row, |
|
|
size_t |
ld_out_col, |
|
|
const void * |
params, |
|
|
unsigned int |
n_channels |
|
) |
| |
|
inlinestatic |
Definition at line 302 of file depthwise_depthfirst.hpp.
310 strat->get_direct_kernel()(
311 n_tile_rows, n_tile_cols,
312 inptr, ld_in_row, ld_in_col,
313 outptr, ld_out_row, ld_out_col,
314 params, n_channels, ws->activation_min, ws->activation_max
◆ indirect()
static void indirect |
( |
const Strat * |
strat, |
|
|
const Workspace * |
ws, |
|
|
const OutputStage & |
, |
|
|
const void * |
params, |
|
|
const TAccum * |
, |
|
|
unsigned int |
n_channels |
|
) |
| |
|
inlinestatic |
◆ supports_direct_kernel
constexpr static bool supports_direct_kernel = true |
|
staticconstexpr |
The documentation for this struct was generated from the following file: