auxiliary data structure used to set attributes to a group of operators More...

#include <op.h>

Collaboration diagram for nnvm::OpGroup:

Public Member Functions
template<typename ValueType >
OpGroup &	set_attr (const std::string &attr_name, const ValueType &value, int plevel=1)
	Register additional attributes to operator group. More...

Detailed Description

auxiliary data structure used to set attributes to a group of operators

Member Function Documentation

template<typename ValueType >

OpGroup & nnvm::OpGroup::set_attr	(	const std::string &	attr_name,
		const ValueType &	value,
		int	plevel = `1`
	)

inline

Register additional attributes to operator group.

Parameters

attr_name	The name of the attribute.
value	The value to be set.
plevel	The priority level of this set, an higher priority level attribute will replace lower priority level attribute. Must be bigger than 0.

Cannot set with same plevel twice in the code.

Template Parameters

ValueType The type of the value to be set.

std::string nnvm::OpGroup::group_name

the tag key to be matched

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