mxnet
|
This is the complete list of members for mxnet::ext::CustomStatefulOp, including all inherited members.
Backward(std::vector< MXTensor > *inputs, std::vector< MXTensor > *outputs, const OpResource &op_res) | mxnet::ext::CustomStatefulOp | inlinevirtual |
create(Ts... args) | mxnet::ext::CustomStatefulOp | inlinestatic |
CustomStatefulOp() | mxnet::ext::CustomStatefulOp | |
Forward(std::vector< MXTensor > *inputs, std::vector< MXTensor > *outputs, const OpResource &op_res)=0 | mxnet::ext::CustomStatefulOp | pure virtual |
ignore_warn | mxnet::ext::CustomStatefulOp | |
wasCreated() | mxnet::ext::CustomStatefulOp | inline |
~CustomStatefulOp() | mxnet::ext::CustomStatefulOp | virtual |