mxnet
|
This is the complete list of members for mxnet::op::SimpleOpRegEntry, including all inherited members.
add_arguments(const std::vector< dmlc::ParamFieldInfo > &args)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
describe(const std::string &description)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
name | mxnet::op::SimpleOpRegEntry | |
set_enable_kwargs(bool enable_kwargs)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_enable_scalar(bool enable_scalar, SimpleOpScalarOption type_mask=kArrayBeforeScalar)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_function(int dev_mask, SourceFunction fsource, SimpleOpRegOption register_symbolic=kRegisterSymbolic)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_function(int dev_mask, UnaryFunction funary, SimpleOpInplaceOption inplace_in_out, SimpleOpRegOption register_symbolic=kRegisterSymbolic)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_function(int dev_mask, BinaryFunction fbinary, SimpleOpInplaceOption inplace_lhs_out, SimpleOpRegOption register_symbolic=kRegisterSymbolic)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_gradient(int dev_mask, UnaryGradFunctionT0 fgrad, SimpleOpInplaceOption inplace_out_in_grad)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_gradient(int dev_mask, UnaryGradFunctionT1 fgrad, SimpleOpInplaceOption inplace_out_in_grad)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_gradient(int dev_mask, UnaryGradFunctionT2 fgrad, SimpleOpInplaceOption inplace_out_in_grad)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_gradient(int dev_mask, BinaryGradFunctionT0 fgrad, SimpleOpInplaceOption inplace_out_lhs_grad)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_gradient(int dev_mask, BinaryGradFunctionT1 fgrad, SimpleOpInplaceOption inplace_out_lhs_grad)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_resource_request(const std::vector< ResourceRequest > &reqs)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_resource_request(ResourceRequest req)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_shape_function(SourceShapeFunction fshapeinfer)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_shape_function(UnaryShapeFunction fshapeinfer)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_shape_function(BinaryShapeFunction fshapeinfer)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
set_symbol_op_name(char const *symbol_name)=0 | mxnet::op::SimpleOpRegEntry | pure virtual |
TSelf typedef | mxnet::op::SimpleOpRegEntry | |
~SimpleOpRegEntry() | mxnet::op::SimpleOpRegEntry | inlinevirtual |