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