ONE - On-device Neural Engine
Loading...
Searching...
No Matches
loco::DepthwiseConv2D Class Referencefinal

Depthwise 2D Convolution. More...

#include <Nodes.h>

Collaboration diagram for loco::DepthwiseConv2D:

Public Member Functions

Nodeifm (void) const
 
void ifm (Node *node)
 
Nodeker (void) const
 
void ker (Node *node)
 
const Padding2Dpad (void) const
 
Padding2Dpad (void)
 
const Stride< 2 > * stride (void) const
 
Stride< 2 > * stride (void)
 
- Public Member Functions inherited from loco::CanonicalNodeDef< CanonicalOpcode::DepthwiseConv2D, FixedArity< 2 >::Mixin >
virtual ~CanonicalNodeDef ()=default
 
uint32_t opnum (void) const final
 
CanonicalOpcode opcode (void) const final
 
- Public Member Functions inherited from loco::CanonicalNode
virtual ~CanonicalNode ()=default
 
const Dialectdialect (void) const final
 Return "Dialect" identifier that this node belongs to.
 
template<typename T >
accept (CanonicalNodeVisitorBase< T > *) const
 
template<typename T >
accept (CanonicalNodeMutableVisitorBase< T > *)
 
- Public Member Functions inherited from loco::Node
 Node ()=default
 
 Node (const Node &)=delete
 
 Node (Node &&)=delete
 
virtual ~Node ()
 
Graphgraph (void)
 
const Graphgraph (void) const
 
virtual uint32_t arity (void) const =0
 Return the number of arguments.
 
virtual Nodearg (uint32_t N) const =0
 Access N-th argument node.
 
virtual void drop (void)=0
 Drop all the reference of arguments.
 
- Public Member Functions inherited from loco::AnnotatedItem< NodeAnnotation >
 AnnotatedItem ()=default
 
virtual ~AnnotatedItem ()=default
 
const T * annot (void) const
 Retrieve a stored annotation of type T.
 
void annot (std::unique_ptr< T > &&p)
 Attach or remove a new annotation of type T.
 

Detailed Description

Depthwise 2D Convolution.

Definition at line 580 of file Nodes.h.

Member Function Documentation

◆ ifm() [1/2]

void loco::DepthwiseConv2D::ifm ( Node node)
inline

Definition at line 585 of file Nodes.h.

585{ at(0)->node(node); }

◆ ifm() [2/2]

Node * loco::DepthwiseConv2D::ifm ( void  ) const
inline

Definition at line 584 of file Nodes.h.

584{ return at(0)->node(); }

Referenced by exo::DepthwiseConv2DConverter::convert(), mir2loco::Transformer::visit(), and mir2loco::Transformer::visit().

◆ ker() [1/2]

void loco::DepthwiseConv2D::ker ( Node node)
inline

Definition at line 588 of file Nodes.h.

588{ at(1)->node(node); }

◆ ker() [2/2]

Node * loco::DepthwiseConv2D::ker ( void  ) const
inline

Definition at line 587 of file Nodes.h.

587{ return at(1)->node(); }

Referenced by exo::DepthwiseConv2DConverter::convert(), and DepthwiseConv2DLayer::operator()().

◆ pad() [1/2]

Padding2D * loco::DepthwiseConv2D::pad ( void  )
inline

Definition at line 592 of file Nodes.h.

592{ return &_pad; }

◆ pad() [2/2]

const Padding2D * loco::DepthwiseConv2D::pad ( void  ) const
inline

Definition at line 591 of file Nodes.h.

591{ return &_pad; }

Referenced by exo::DepthwiseConv2DConverter::convert().

◆ stride() [1/2]

Stride< 2 > * loco::DepthwiseConv2D::stride ( void  )
inline

Definition at line 596 of file Nodes.h.

596{ return &_stride; }

◆ stride() [2/2]

const Stride< 2 > * loco::DepthwiseConv2D::stride ( void  ) const
inline

Definition at line 595 of file Nodes.h.

595{ return &_stride; }

Referenced by exo::DepthwiseConv2DConverter::convert().


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