InputLayer::Return Class Reference

#include <GraphBuilder.h>

Public Member Functions
	Return (loco::GraphInput *input, loco::Pull *node)
loco::Pull *	node (void)
Return *	name (const std::string &value)
Return *	shape (std::initializer_list< uint32_t > dims)

Detailed Description

Definition at line 115 of file GraphBuilder.h.

Constructor & Destructor Documentation

Return()

InputLayer::Return::Return ( loco::GraphInput *  input, loco::Pull *  node  )

inline

Definition at line 118 of file GraphBuilder.h.

                                                  : _input{input}, _node{node}
    {
      // DO NOTHING
    }

Member Function Documentation

name()

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

inline

Definition at line 127 of file GraphBuilder.h.

    {
      _input->name(value);
      return this;
    }

node()

loco::Pull * InputLayer::Return::node ( void  )

inline

Definition at line 124 of file GraphBuilder.h.

{ return _node; }

shape()

Return * InputLayer::Return::shape ( std::initializer_list< uint32_t >  dims )

inline

Definition at line 134 of file GraphBuilder.h.

    {
      // TODO Uncomment this line when GraphInput is ready
      // _graph_input->shape(dims)
      _node->shape(dims);
      return this;
    }

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

---

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

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