locoex::TFLTransposeConv Class Reference

TRANSPOSE_CONV in TensorFlow Lite. More...

#include <TFLNodes.h>

Collaboration diagram for locoex::TFLTransposeConv:

Public Member Functions
loco::Node *	inputSizes (void) const
void	inputSizes (Node *node)
loco::Node *	filter (void) const
void	filter (Node *node)
loco::Node *	outBackprop (void) const
void	outBackprop (Node *node)
const Padding &	padding (void) const
void	padding (const Padding &padding)
const Stride *	stride (void) const
Stride *	stride (void)
Public Member Functions inherited from locoex::FixedArityNode< 3, TFLNodeImpl< TFLOpcode::TRANSPOSE_CONV > >
	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::TRANSPOSE_CONV >
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.

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

Detailed Description

TRANSPOSE_CONV in TensorFlow Lite.

Note

Argument node function names are from TensorFlow. So refering 'in' and 'out' acutally means 'out' and 'in' of the this node.

Definition at line 527 of file TFLNodes.h.

Member Function Documentation

filter() [1/2]

void locoex::TFLTransposeConv::filter ( Node *  node )

inline

Definition at line 534 of file TFLNodes.h.

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

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

filter() [2/2]

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

inline

Definition at line 533 of file TFLNodes.h.

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

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

inputSizes() [1/2]

void locoex::TFLTransposeConv::inputSizes ( Node *  node )

inline

Definition at line 531 of file TFLNodes.h.

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

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

inputSizes() [2/2]

loco::Node * locoex::TFLTransposeConv::inputSizes ( void  ) const

inline

Definition at line 530 of file TFLNodes.h.

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

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

outBackprop() [1/2]

void locoex::TFLTransposeConv::outBackprop ( Node *  node )

inline

Definition at line 537 of file TFLNodes.h.

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

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

outBackprop() [2/2]

loco::Node * locoex::TFLTransposeConv::outBackprop ( void  ) const

inline

Definition at line 536 of file TFLNodes.h.

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

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

padding() [1/2]

void locoex::TFLTransposeConv::padding ( const Padding &  padding )

inline

Definition at line 541 of file TFLNodes.h.

{ _padding = padding; }

References padding().

Referenced by padding().

padding() [2/2]

const Padding & locoex::TFLTransposeConv::padding ( void  ) const

inline

Definition at line 540 of file TFLNodes.h.

{ return _padding; }

stride() [1/2]

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

inline

Definition at line 544 of file TFLNodes.h.

{ return &_stride; }

stride() [2/2]

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

inline

Definition at line 543 of file TFLNodes.h.

{ return &_stride; }

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

---

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

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