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.
- パラメータ:
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.
- パラメータ:
inputs -- [in] Function inputs as CgVariables.
-
inline CgVariablePtr output(int i)
Get a weak reference output as a shared reference by index or raise.
- パラメータ:
i -- [in] Output index.
- NBLA_API void set_outputs (const vector< CgVariablePtr > &outputs)
Store outputs as weak references (weak_ptr).
- パラメータ:
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.
-
NBLA_API CgFunction(FunctionPtr func)