mxnet
|
Registry entry for NDArrayFunction. More...
#include <ndarray.h>
Public Member Functions | |
NDArrayFunctionReg () | |
constructor More... | |
NDArrayFunctionReg & | set_function (void(*fsetvalue)(const real_t &rhs, NDArray *out)) |
set the function body to a NDArray setvalue function this will also auto set the parameters correctly More... | |
NDArrayFunctionReg & | set_function (void(*fternary)(const NDArray &lhs, const NDArray &mhs, const NDArray &rhs, NDArray *out)) |
set the function body to a ternary NDArray function this will also auto set the parameters correctly More... | |
NDArrayFunctionReg & | set_function (void(*fbinary)(const NDArray &lhs, const NDArray &rhs, NDArray *out)) |
set the function body to a binary NDArray function this will also auto set the parameters correctly More... | |
NDArrayFunctionReg & | set_function (void(*fscalar)(const NDArray &lhs, const real_t &rhs, NDArray *out)) |
set the function body to a binary NDArray function this will also auto set the parameters correctly More... | |
NDArrayFunctionReg & | set_function (void(*funary)(const NDArray &src, NDArray *out)) |
set the function body to a unary NDArray function this will also auto set the parameters correctly More... | |
NDArrayFunctionReg & | set_function (void(*fgeneric)(NDArray **used_vars, real_t *s, NDArray **mutate_vars, const std::map< std::string, std::string > ¶m)) |
set the function body to a unary NDArray function this will also auto set the parameters correctly More... | |
NDArrayFunctionReg & | set_num_use_vars (unsigned n) |
set the number of mutate variables More... | |
NDArrayFunctionReg & | set_num_mutate_vars (unsigned n) |
set the number of mutate variables More... | |
NDArrayFunctionReg & | set_num_scalars (unsigned n) |
set the number of scalar arguments More... | |
NDArrayFunctionReg & | set_type_mask (int tmask) |
set type mask More... | |
Public Member Functions inherited from dmlc::FunctionRegEntryBase< NDArrayFunctionReg, NDArrayAPIFunction > | |
NDArrayFunctionReg & | set_body (NDArrayAPIFunctionbody) |
Set the function body. More... | |
NDArrayFunctionReg & | describe (const std::string &description) |
Describe the function. More... | |
NDArrayFunctionReg & | add_argument (const std::string &name, const std::string &type, const std::string &description) |
Add argument information to the function. More... | |
NDArrayFunctionReg & | add_arguments (const std::vector< ParamFieldInfo > &args) |
Append list if arguments to the end. More... | |
NDArrayFunctionReg & | set_return_type (const std::string &type) |
Set the return type. More... | |
Public Attributes | |
unsigned | num_use_vars |
number of variable used by this function More... | |
unsigned | num_mutate_vars |
number of variable mutated by this function More... | |
unsigned | num_scalars |
number of scalars used by this function More... | |
int | type_mask |
information on how function should be called from API More... | |
Public Attributes inherited from dmlc::FunctionRegEntryBase< NDArrayFunctionReg, NDArrayAPIFunction > | |
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... | |
NDArrayAPIFunction | body |
Function body to create ProductType. More... | |
std::string | return_type |
Return type of the function. More... | |
Additional Inherited Members | |
Protected Member Functions inherited from dmlc::FunctionRegEntryBase< NDArrayFunctionReg, NDArrayAPIFunction > | |
NDArrayFunctionReg & | self () |
Registry entry for NDArrayFunction.
|
inline |
constructor
|
inline |
set the function body to a NDArray setvalue function this will also auto set the parameters correctly
fsetvalue | function body to set |
|
inline |
set the function body to a ternary NDArray function this will also auto set the parameters correctly
fternary | function body to set |
|
inline |
set the function body to a binary NDArray function this will also auto set the parameters correctly
fbinary | function body to set |
|
inline |
set the function body to a binary NDArray function this will also auto set the parameters correctly
fscalar | function body to set |
|
inline |
set the function body to a unary NDArray function this will also auto set the parameters correctly
funary | function body to set |
|
inline |
set the function body to a unary NDArray function this will also auto set the parameters correctly
fgeneric | function body to set |
|
inline |
set the number of mutate variables
n | number of mutate variablesx |
|
inline |
set the number of scalar arguments
n | number of scalar arguments |
|
inline |
set the number of mutate variables
n | number of mutate variablesx |
|
inline |
set type mask
tmask | typemask |
unsigned mxnet::NDArrayFunctionReg::num_mutate_vars |
number of variable mutated by this function
unsigned mxnet::NDArrayFunctionReg::num_scalars |
number of scalars used by this function
unsigned mxnet::NDArrayFunctionReg::num_use_vars |
number of variable used by this function
int mxnet::NDArrayFunctionReg::type_mask |
information on how function should be called from API