InputPatchElement< T, IsGeneric, OutputStage > Class Template Reference

#include <depthwise_depthfirst_multiplier.hpp>

Data Structures
struct	Workspace

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

Detailed Description

template<typename T, bool IsGeneric = false, typename OutputStage = Nothing>
class arm_conv::depthwise::depthfirst_multiplier::InputPatchElement< T, IsGeneric, OutputStage >

Definition at line 255 of file depthwise_depthfirst_multiplier.hpp.

Member Function Documentation

get_element_size()

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

inlinestatic

Definition at line 266 of file depthwise_depthfirst_multiplier.hpp.

  {
    return sizeof_input_rows(args) + sizeof_input_padding(args) + sizeof_input_patch(args);
  }

References GemmTuner::args.

initialise()

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

inlinestatic

Definition at line 272 of file depthwise_depthfirst_multiplier.hpp.

  {
    auto buffer_bytes = reinterpret_cast<char *>(buffer);
 
    ws->input_rows = reinterpret_cast<const T **>(buffer_bytes);
    buffer_bytes += sizeof_input_rows(args);
 
    ws->input_padding = reinterpret_cast<T*>(buffer_bytes);
    buffer_bytes += sizeof_input_padding(args);
 
    ws->input_patch = reinterpret_cast<T*>(buffer_bytes);
    buffer_bytes += sizeof_input_patch(args);
 
    // Initialise the padding
    memset(ws->input_padding,
           get_input_buffer_fill_value(args.output_stage),
           sizeof_input_padding(args));
 
    return buffer_bytes;
  }

References GemmTuner::args.

---

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

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