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

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

inlinevirtual

virtual MXReturnValue 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