locoex::TFLReshape Class Reference

#include <TFLNodes.h>

Collaboration diagram for locoex::TFLReshape:

Data Structures
class	Shape

Public Member Functions
	TFLReshape ()=default
loco::Node *	tensor (void) const
void	tensor (loco::Node *node)
loco::Node *	shape (void) const
void	shape (loco::Node *node)
const Shape *	newShape (void) const
Shape *	newShape (void)
Public Member Functions inherited from locoex::FixedArityNode< 2, TFLNodeImpl< TFLOpcode::RESHAPE > >
	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::RESHAPE >
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< 2, TFLNodeImpl< TFLOpcode::RESHAPE > >
loco::Use *	at (unsigned n) const

Detailed Description

Definition at line 404 of file TFLNodes.h.

Constructor & Destructor Documentation

TFLReshape()

locoex::TFLReshape::TFLReshape ( )

default

Member Function Documentation

newShape() [1/2]

Shape * locoex::TFLReshape::newShape ( void  )

inline

Definition at line 433 of file TFLNodes.h.

{ return &_new_shape; }

newShape() [2/2]

const Shape * locoex::TFLReshape::newShape ( void  ) const

inline

Definition at line 432 of file TFLNodes.h.

{ return &_new_shape; }

Referenced by locoex::set_new_shape().

shape() [1/2]

void locoex::TFLReshape::shape ( loco::Node *  node )

inline

Definition at line 416 of file TFLNodes.h.

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

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

Referenced by RandomDataGenerator.RandomDataGenerator::_gen_float32(), RandomDataGenerator.RandomDataGenerator::_gen_int16(), and RandomDataGenerator.RandomDataGenerator::_gen_uint8().

shape() [2/2]

loco::Node * locoex::TFLReshape::shape ( void  ) const

inline

Definition at line 415 of file TFLNodes.h.

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

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

Referenced by RandomDataGenerator.RandomDataGenerator::_gen_float32(), RandomDataGenerator.RandomDataGenerator::_gen_int16(), RandomDataGenerator.RandomDataGenerator::_gen_uint8(), and locoex::set_new_shape().

tensor() [1/2]

void locoex::TFLReshape::tensor ( loco::Node *  node )

inline

Definition at line 411 of file TFLNodes.h.

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

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

tensor() [2/2]

loco::Node * locoex::TFLReshape::tensor ( void  ) const

inline

Definition at line 410 of file TFLNodes.h.

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

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

---

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

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