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>

Collaboration diagram for mxnet::ext::CustomStatefulOp:

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

Detailed Description

An abstract class for library authors creating stateful op custom library should override Forward and destructor, and has an option to implement Backward.

Constructor & Destructor Documentation

◆ CustomStatefulOp()

mxnet::ext::CustomStatefulOp::CustomStatefulOp ( )

◆ ~CustomStatefulOp()

virtual mxnet::ext::CustomStatefulOp::~CustomStatefulOp ( )

virtual

Member Function Documentation

◆ Backward()

virtual MXReturnValue mxnet::ext::CustomStatefulOp::Backward	(	std::vector< MXTensor > *	inputs,
		std::vector< MXTensor > *	outputs,
		const OpResource &	op_res
	)

inlinevirtual

◆ create()

template<class A , typename ... Ts>

static CustomStatefulOp* mxnet::ext::CustomStatefulOp::create ( Ts... args )

inlinestatic

◆ Forward()

virtual MXReturnValue mxnet::ext::CustomStatefulOp::Forward	(	std::vector< MXTensor > *	inputs,
		std::vector< MXTensor > *	outputs,
		const OpResource &	op_res
	)

pure virtual

◆ wasCreated()

bool mxnet::ext::CustomStatefulOp::wasCreated ( )

inline

Member Data Documentation

◆ ignore_warn

bool mxnet::ext::CustomStatefulOp::ignore_warn

The documentation for this class was generated from the following file:

/work/mxnet/include/mxnet/lib_api.h

Public Member Functions

Static Public Member Functions

Public Attributes