mxnet::ext::CustomStatefulOpWrapper Class Reference

StatefulOp wrapper class to pass to backend OpState. More...

#include <lib_api.h>

Collaboration diagram for mxnet::ext::CustomStatefulOpWrapper:

Public Member Functions
	~CustomStatefulOpWrapper ()
	CustomStatefulOpWrapper (CustomStatefulOp *inst, opCallDestroyOpState_t destroy)
CustomStatefulOp *	get_instance ()

Detailed Description

StatefulOp wrapper class to pass to backend OpState.

Constructor & Destructor Documentation

~CustomStatefulOpWrapper()

mxnet::ext::CustomStatefulOpWrapper::~CustomStatefulOpWrapper

(

)

CustomStatefulOpWrapper()

mxnet::ext::CustomStatefulOpWrapper::CustomStatefulOpWrapper ( CustomStatefulOp *  inst, opCallDestroyOpState_t  destroy  )

inlineexplicit

Member Function Documentation

get_instance()

CustomStatefulOp* mxnet::ext::CustomStatefulOpWrapper::get_instance ( )

inline

---

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

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