OutputLayer::Return Class Reference

#include <GraphBuilder.h>

Public Member Functions
	Return (loco::GraphOutput *output, loco::Push *node)
loco::Push *	node (void)
Return *	name (const std::string &value)

Detailed Description

Definition at line 167 of file GraphBuilder.h.

Constructor & Destructor Documentation

Return()

OutputLayer::Return::Return ( loco::GraphOutput *  output, loco::Push *  node  )

inline

Definition at line 170 of file GraphBuilder.h.

                                                    : _output{output}, _node{node}
    {
      // DO NOTHING
    }

Member Function Documentation

name()

Return * OutputLayer::Return::name ( const std::string &  value )

inline

Definition at line 179 of file GraphBuilder.h.

    {
      // TODO Uncomment this line when GraphOutput is ready
      // _graph_output->shape(dims)
      _output->name(value);
      return this;
    }

node()

loco::Push * OutputLayer::Return::node ( void  )

inline

Definition at line 176 of file GraphBuilder.h.

{ return _node; }

---

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

• compiler/loco/src/Service/GraphBuilder.h