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 | |
CustomStatefulOp () | |
virtual | ~CustomStatefulOp () |
bool | wasCreated () |
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) |
Static Public Member Functions | |
template<class A , typename ... Ts> | |
static CustomStatefulOp * | create (Ts...args) |
Public Attributes | |
bool | ignore_warn |
An abstract class for library authors creating stateful op custom library should override Forward and destructor, and has an option to implement Backward.
mxnet::ext::CustomStatefulOp::CustomStatefulOp | ( | ) |
|
virtual |
|
inlinevirtual |
|
inlinestatic |
|
pure virtual |
|
inline |
bool mxnet::ext::CustomStatefulOp::ignore_warn |