class nbla::BaseTransformUnary

template<typename ...Args> class BaseTransformUnary : public nbla::BaseFunction<Args...>

Public Functions

inline virtual int min_inputs()

Get minimum number of inputs.

This is meant to be used in setup function with in_types which is used to get maximum number of inputs.

inline virtual int min_outputs()

Get minimum number of outputs.

This is meant to be used in setup function with out_types which is used to get max number of outputs.

inline virtual int inplace_data(int i) const

Get in-place-level of i-th input variable’s data (see below).

0 (NOT_INPLACE): Not in-placed 1 (INPLACE_NOT_MODIFY): In-placed but not modified. 2 (INPLACE): In-placed and modified.

Note

If a subclass uses in-place computation, the function must override this function.

inline virtual int inplace_data_with(int i) const

Get the output variable index where i-th variables’ data in-placed to.

Note

This is only valid if the i-th variable is in-placed. The maintainer of a sub-class function must override this function.