Registry entry for OperatorProperty factory functions.
More...
#include <operator.h>
Registry entry for OperatorProperty factory functions.
Check if TypeString of the type matches the registered name.
OperatorPropertyReg& mxnet::OperatorPropertyReg::set_key_var_num_args |
( |
const std::string & |
key | ) |
|
|
inline |
Set key_var_num_args When this is set, the API caller is required to pass in a argument with key=key_num_args.c_str(), and value=num_args. num_args is number of positional argument when calling the function.
This is used to pass in length of positional arguments for operators that can take variable length of input. Most operators do not need to set this property.
- Parameters
-
key | the key name to be set |
std::string mxnet::OperatorPropertyReg::key_var_num_args |
The documentation for this struct was generated from the following file: