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
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)

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.

Member Function Documentation

◆ Backward()

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

inlinevirtual

◆ Forward()

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

pure virtual

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

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

Public Member Functions

Detailed Description

Member Function Documentation

◆ Backward()

◆ Forward()