mxnet
Public Member Functions | Public Attributes | List of all members
mxnet::OperatorPropertyReg Struct Reference

Registry entry for OperatorProperty factory functions. More...

#include <operator.h>

Inheritance diagram for mxnet::OperatorPropertyReg:
Inheritance graph
Collaboration diagram for mxnet::OperatorPropertyReg:
Collaboration graph

Public Member Functions

OperatorPropertyRegset_key_var_num_args (const std::string &key)
 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. More...
 
OperatorPropertyRegcheck_name ()
 Check if TypeString of the type matches the registered name. More...
 

Public Attributes

std::string key_var_num_args
 The key num_args name. More...
 

Detailed Description

Registry entry for OperatorProperty factory functions.

Member Function Documentation

OperatorPropertyReg& mxnet::OperatorPropertyReg::check_name ( )
inline

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
keythe key name to be set

Member Data Documentation

std::string mxnet::OperatorPropertyReg::key_var_num_args

The key num_args name.


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