locoex::TFLDepthwiseConv2D Class Reference

DEPTHWISE_CONV_2D in TensorFlow Lite. More...

#include <TFLNodes.h>

Collaboration diagram for locoex::TFLDepthwiseConv2D:

Public Member Functions
loco::Node *	input (void) const
void	input (loco::Node *node)
loco::Node *	filter (void) const
void	filter (loco::Node *node)
loco::Node *	bias (void) const override
void	bias (loco::Node *node) override
	get the input for bias.
Padding	padding () const
void	padding (Padding padding)
const Stride *	stride (void) const
Stride *	stride (void)
int32_t	depthMultiplier (void) const
void	depthMultiplier (int32_t arg)
Public Member Functions inherited from locoex::FixedArityNode< 3, TFLNodeImpl< TFLOpcode::DEPTHWISE_CONV_2D > >
	FixedArityNode ()
virtual	~FixedArityNode ()=default
unsigned	arity (void) const final
loco::Node *	arg (uint32_t n) const final
void	drop (void) final
Public Member Functions inherited from locoex::TFLNodeImpl< TFLOpcode::DEPTHWISE_CONV_2D >
virtual	~TFLNodeImpl ()=default
uint32_t	opnum (void) const final
TFLOpcode	opcode (void) const final
Public Member Functions inherited from locoex::TFLNode
virtual	~TFLNode ()=default
const loco::Dialect *	dialect (void) const final
	Return "Dialect" identifier that this node belongs to.
template<typename T >
T	accept (TFLNodeVisitorBase< T > *) const
template<typename T >
T	accept (TFLNodeMutableVisitorBase< T > *)
Public Member Functions inherited from loco::Node
	Node ()=default
	Node (const Node &)=delete
	Node (Node &&)=delete
virtual	~Node ()
Graph *	graph (void)
const Graph *	graph (void) const
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.
Public Member Functions inherited from locoex::TFLNodeMixin< TFLNodeTrait::FusedActFunc >
	TFLNodeMixin ()=default
FusedActFunc	fusedActivationFunction () const
void	fusedActivationFunction (FusedActFunc fused_act_fun)
Public Member Functions inherited from locoex::TFLNodeMixin< TFLNodeTrait::Bias >
	TFLNodeMixin ()=default

Additional Inherited Members
Protected Member Functions inherited from locoex::FixedArityNode< 3, TFLNodeImpl< TFLOpcode::DEPTHWISE_CONV_2D > >
loco::Use *	at (unsigned n) const

Detailed Description

DEPTHWISE_CONV_2D in TensorFlow Lite.

Definition at line 244 of file TFLNodes.h.

Member Function Documentation

bias() [1/2]

void locoex::TFLDepthwiseConv2D::bias ( loco::Node *  node )

inlineoverridevirtual

get the input for bias.

Implements locoex::TFLNodeMixin< TFLNodeTrait::Bias >.

Definition at line 257 of file TFLNodes.h.

{ at(2)->node(node); }

References locoex::FixedArityNode< 3, TFLNodeImpl< TFLOpcode::DEPTHWISE_CONV_2D > >::at(), and loco::Use::node().

bias() [2/2]

loco::Node * locoex::TFLDepthwiseConv2D::bias ( void  ) const

inlineoverridevirtual

Implements locoex::TFLNodeMixin< TFLNodeTrait::Bias >.

Definition at line 256 of file TFLNodes.h.

{ return at(2)->node(); }

References locoex::FixedArityNode< 3, TFLNodeImpl< TFLOpcode::DEPTHWISE_CONV_2D > >::at(), and loco::Use::node().

depthMultiplier() [1/2]

void locoex::TFLDepthwiseConv2D::depthMultiplier ( int32_t  arg )

inline

Definition at line 267 of file TFLNodes.h.

{ _depth_multiplier = arg; }

References locoex::FixedArityNode< 3, TFLNodeImpl< TFLOpcode::DEPTHWISE_CONV_2D > >::arg().

depthMultiplier() [2/2]

int32_t locoex::TFLDepthwiseConv2D::depthMultiplier ( void  ) const

inline

Definition at line 266 of file TFLNodes.h.

{ return _depth_multiplier; }

filter() [1/2]

void locoex::TFLDepthwiseConv2D::filter ( loco::Node *  node )

inline

Definition at line 254 of file TFLNodes.h.

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

References locoex::FixedArityNode< 3, TFLNodeImpl< TFLOpcode::DEPTHWISE_CONV_2D > >::at(), and loco::Use::node().

filter() [2/2]

loco::Node * locoex::TFLDepthwiseConv2D::filter ( void  ) const

inline

Definition at line 253 of file TFLNodes.h.

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

References locoex::FixedArityNode< 3, TFLNodeImpl< TFLOpcode::DEPTHWISE_CONV_2D > >::at(), and loco::Use::node().

input() [1/2]

void locoex::TFLDepthwiseConv2D::input ( loco::Node *  node )

inline

Definition at line 251 of file TFLNodes.h.

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

References locoex::FixedArityNode< 3, TFLNodeImpl< TFLOpcode::DEPTHWISE_CONV_2D > >::at(), and loco::Use::node().

input() [2/2]

loco::Node * locoex::TFLDepthwiseConv2D::input ( void  ) const

inline

Definition at line 250 of file TFLNodes.h.

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

References locoex::FixedArityNode< 3, TFLNodeImpl< TFLOpcode::DEPTHWISE_CONV_2D > >::at(), and loco::Use::node().

padding() [1/2]

Padding locoex::TFLDepthwiseConv2D::padding ( ) const

inline

Definition at line 260 of file TFLNodes.h.

{ return _padding; }

padding() [2/2]

void locoex::TFLDepthwiseConv2D::padding ( Padding  padding )

inline

Definition at line 261 of file TFLNodes.h.

{ _padding = padding; }

References padding().

Referenced by padding().

stride() [1/2]

Stride * locoex::TFLDepthwiseConv2D::stride ( void  )

inline

Definition at line 264 of file TFLNodes.h.

{ return &_stride; }

stride() [2/2]

const Stride * locoex::TFLDepthwiseConv2D::stride ( void  ) const

inline

Definition at line 263 of file TFLNodes.h.

{ return &_stride; }

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

---

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

• compiler/exo/src/Dialect/IR/TFLNodes.h