class nbla::CgFunction

class CgFunction

Computation graph function.

A Function object is held in this object, and pointers to inputs and outputs also kept.

Public Functions

NBLA_API CgFunction(FunctionPtr func)

Ctor.

Parameters:

func – [in] shared_ptr of Function.

NBLA_API ~CgFunction()

Dtor.

Erase all function_reference_ of inputs.

inline void set_inputs_(const vector<CgVariablePtr> &inputs)

Set inputs.

Note user shouldn’t call this directly.

Parameters:

inputs – [in] Function inputs as CgVariables.

void setup()

Calling setup function of an Function object internally held.

inline CgVariablePtr output(int i)

Get a weak reference output as a shared reference by index or raise.

Parameters:

i – [in] Output index.

NBLA_API void set_outputs (const vector< CgVariablePtr > &outputs)

Store outputs as weak references (weak_ptr).

Parameters:

outputs – [in] Function outputs.

inline size_t num_inputs() const

Get number of inputs.

inline CgVariablePtr input(size_t i)

Get the i-th input.

inline size_t num_outputs() const

Get number of outputs.