Compute Library
 22.08
GenericInputArrayElement< T > Class Template Reference

#include <depthwise_depthfirst_generic.hpp>

Data Structures

struct  Workspace
 

Static Public Member Functions

template<class OutputStage >
static size_t get_element_size (const WorkspaceArgs< IDepthfirstStrategy, OutputStage > &args)
 
template<class WorkspaceType , class OutputStage >
static void * initialise (WorkspaceType *ws, void *buffer, const WorkspaceArgs< IDepthfirstStrategy, OutputStage > &args)
 

Detailed Description

template<typename T>
class arm_conv::depthwise::GenericInputArrayElement< T >

Definition at line 177 of file depthwise_depthfirst_generic.hpp.

Member Function Documentation

◆ get_element_size()

static size_t get_element_size ( const WorkspaceArgs< IDepthfirstStrategy, OutputStage > &  args)
inlinestatic

Definition at line 186 of file depthwise_depthfirst_generic.hpp.

187  {
188  const auto kernel_points = args.depthwise_args.kernel_rows * args.depthwise_args.kernel_cols;
189  return sizeof(T **) * args.strategy->get_input_rows() * args.strategy->get_input_cols() * kernel_points;
190  }

◆ initialise()

static void* initialise ( WorkspaceType *  ws,
void *  buffer,
const WorkspaceArgs< IDepthfirstStrategy, OutputStage > &  args 
)
inlinestatic

Definition at line 193 of file depthwise_depthfirst_generic.hpp.

194  {
195  ws->inptr_array = reinterpret_cast<const T**>(buffer);
196  return reinterpret_cast<char *>(buffer) + get_element_size(args);
197  }
static size_t get_element_size(const WorkspaceArgs< IDepthfirstStrategy, OutputStage > &args)

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