mxnet
|
An abstract class for library authors creating stateful op custom library should override Forward and destructor, and has an option to implement Backward. More...
#include <lib_api.h>
Public Member Functions | |
virtual MXReturnValue | Forward (std::vector< MXTensor > *inputs, std::vector< MXTensor > *outputs, const OpResource &op_res)=0 |
virtual MXReturnValue | Backward (std::vector< MXTensor > *inputs, std::vector< MXTensor > *outputs, const OpResource &op_res) |
An abstract class for library authors creating stateful op custom library should override Forward and destructor, and has an option to implement Backward.
|
inlinevirtual |
|
pure virtual |