moco::TFSoftmax Class Reference

#include <TFSoftmax.h>

Collaboration diagram for moco::TFSoftmax:

Public Member Functions
	TFSoftmax ()=default
Node *	logits (void) const
void	logits (Node *node)
Public Member Functions inherited from moco::FixedArityNode< 1, TFNodeImpl< TFOpcode::Softmax > >
	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 moco::TFNodeImpl< TFOpcode::Softmax >
virtual	~TFNodeImpl ()=default
uint32_t	opnum (void) const final
TFOpcode	opcode (void) const final
Public Member Functions inherited from moco::TFNode
virtual	~TFNode ()=default
const loco::Dialect *	dialect (void) const final
	Return "Dialect" identifier that this node belongs to.
template<typename T >
T	accept (TFNodeVisitorBase< T > *) const
template<typename T >
T	accept (TFNodeMutableVisitorBase< T > *)
NodeName	name (void) const
void	name (const NodeName &name)
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 moco::FixedArityNode< 1, TFNodeImpl< TFOpcode::Softmax > >
loco::Use *	at (unsigned n) const

Detailed Description

Definition at line 25 of file TFSoftmax.h.

Constructor & Destructor Documentation

TFSoftmax()

moco::TFSoftmax::TFSoftmax ( )

default

Member Function Documentation

logits() [1/2]

void moco::TFSoftmax::logits ( Node *  node )

inline

Definition at line 32 of file TFSoftmax.h.

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

References moco::FixedArityNode< 1, TFNodeImpl< TFOpcode::Softmax > >::at(), and loco::Use::node().

logits() [2/2]

Node * moco::TFSoftmax::logits ( void  ) const

inline

Definition at line 31 of file TFSoftmax.h.

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

References moco::FixedArityNode< 1, TFNodeImpl< TFOpcode::Softmax > >::at(), and loco::Use::node().

---

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

• compiler/moco/lang/include/moco/IR/Nodes/TFSoftmax.h