mxnet
|
Common base class for function registry. More...
#include <registry.h>
Public Member Functions | |
EntryType & | set_body (FunctionType body) |
Set the function body. More... | |
EntryType & | describe (const std::string &description) |
Describe the function. More... | |
EntryType & | add_argument (const std::string &name, const std::string &type, const std::string &description) |
Add argument information to the function. More... | |
EntryType & | add_arguments (const std::vector< ParamFieldInfo > &args) |
Append list if arguments to the end. More... | |
EntryType & | set_return_type (const std::string &type) |
Set the return type. More... | |
Public Attributes | |
std::string | name |
name of the entry More... | |
std::string | description |
description of the entry More... | |
std::vector< ParamFieldInfo > | arguments |
additional arguments to the factory function More... | |
FunctionType | body |
Function body to create ProductType. More... | |
std::string | return_type |
Return type of the function. More... | |
Protected Member Functions | |
EntryType & | self () |
Common base class for function registry.
EntryType | The type of subclass that inheritate the base. |
FunctionType | The function type this registry is registerd. |
|
inline |
Add argument information to the function.
name | Name of the argument. |
type | Type of the argument. |
description | Description of the argument. |
|
inline |
Append list if arguments to the end.
args | Additional list of arguments. |
|
inline |
Describe the function.
description | The description of the factory function. |
|
inlineprotected |
|
inline |
Set the function body.
body | Function body to set. |
|
inline |
Set the return type.
type | Return type of the function, could be Symbol or Symbol[] |
std::vector<ParamFieldInfo> dmlc::FunctionRegEntryBase< EntryType, FunctionType >::arguments |
additional arguments to the factory function
FunctionType dmlc::FunctionRegEntryBase< EntryType, FunctionType >::body |
Function body to create ProductType.
std::string dmlc::FunctionRegEntryBase< EntryType, FunctionType >::description |
description of the entry
std::string dmlc::FunctionRegEntryBase< EntryType, FunctionType >::name |
name of the entry
std::string dmlc::FunctionRegEntryBase< EntryType, FunctionType >::return_type |
Return type of the function.