docc 0.8.3 (0d44999b)
C++ API Reference
Loading...
Searching...
No Matches
sdfg::offloading::GPUConvExpander Class Reference

#include <sdfg/targets/gpu/math/tensor/conv_expander.h>

Inheritance diagram for sdfg::offloading::GPUConvExpander:
[legend]

Public Member Functions

 GPUConvExpander ()
virtual passes::LibNodeExpander::ExpandOutcome handle_expand (passes::LibNodeExpander::ExpandContext &context, structured_control_flow::Block &block, math::tensor::ConvNode &node) const override
Public Member Functions inherited from sdfg::passes::CodeLibNodeExpander< math::tensor::ConvNode >
 CodeLibNodeExpander (const data_flow::LibraryNodeCode &code)
const data_flow::LibraryNodeCode & code () const
 The fixed library node code this expander dispatches on.
const data_flow::LibraryNodeCode * matched_code () const override
 If this expander dispatches on a single, fixed library node code, expose it here.
const LibNodeExpander * for_lib_node (const data_flow::LibraryNode &node) const override
 Prefiltering of libnodes.
Public Member Functions inherited from sdfg::passes::TypedLibNodeExpander< math::tensor::ConvNode >
const LibNodeExpander * for_lib_node (const data_flow::LibraryNode &node) const override
 Prefiltering of libnodes.
Public Member Functions inherited from sdfg::passes::LibNodeExpander
virtual ~LibNodeExpander ()=default
virtual ExpandOutcome handle_expand (ExpandContext &context, structured_control_flow::Block &block, data_flow::LibraryNode &node) const =0

Static Public Member Functions

static passes::LibNodeExpander::ExpandOutcome handle_expand_im2row (passes::LibNodeExpander::ExpandContext &context, structured_control_flow::Block &block, math::tensor::ConvNode &node)

Additional Inherited Members

Public Types inherited from sdfg::passes::LibNodeExpander
enum class  InputUse {
  Skip = 0 , Scalar , IndirectRead , IndirectWrite ,
  IndirectReadWrite
}

Constructor & Destructor Documentation

◆ GPUConvExpander()

sdfg::offloading::GPUConvExpander::GPUConvExpander ( )
inline

Member Function Documentation

◆ handle_expand()

virtual passes::LibNodeExpander::ExpandOutcome sdfg::offloading::GPUConvExpander::handle_expand ( passes::LibNodeExpander::ExpandContext & context,
structured_control_flow::Block & block,
math::tensor::ConvNode & node ) const
overridevirtual

◆ handle_expand_im2row()

passes::LibNodeExpander::ExpandOutcome sdfg::offloading::GPUConvExpander::handle_expand_im2row ( passes::LibNodeExpander::ExpandContext & context,
structured_control_flow::Block & block,
math::tensor::ConvNode & node )
static

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