Class to hold custom operator registration. More...

#include <lib_api.h>

Collaboration diagram for mxnet::ext::CustomOp:

Public Member Functions
	CustomOp (const char *op_name)

CustomOp &	setForward (fcomp_t fcomp, const char *ctx)

CustomOp &	setBackward (fcomp_t fgrad, const char *ctx)

CustomOp &	setParseAttrs (parseAttrs_t func)

CustomOp &	setInferType (inferType_t func)

CustomOp &	setInferSType (inferSType_t func)

CustomOp &	setInferShape (inferShape_t func)

CustomOp &	setMutateInputs (mutateInputs_t func)

CustomOp &	setCreateOpState (createOpState_t func, const char *ctx)

CustomOp &	setIsSubgraphOp ()

void	mapToVector ()

Public Attributes
const char *	name
	operator name More...

parseAttrs_t	parse_attrs
	operator functions More...

inferType_t	infer_type

inferSType_t	infer_storage_type

inferShape_t	infer_shape

mutateInputs_t	mutate_inputs

bool	isSGop

std::vector< const char * >	forward_ctx_cstr
	vector repr of ctx map to be easily loaded from c_api More...

std::vector< const char * >	backward_ctx_cstr

std::vector< const char * >	create_op_ctx_cstr

std::vector< fcomp_t >	forward_fp

std::vector< fcomp_t >	backward_fp

std::vector< createOpState_t >	create_op_fp

Detailed Description

Class to hold custom operator registration.

Constructor & Destructor Documentation

◆ CustomOp()

mxnet::ext::CustomOp::CustomOp ( const char * op_name )

explicit

Member Function Documentation

◆ mapToVector()

void mxnet::ext::CustomOp::mapToVector ( )

◆ setBackward()

CustomOp& mxnet::ext::CustomOp::setBackward	(	fcomp_t	fgrad,
		const char *	ctx
	)

◆ setCreateOpState()

CustomOp& mxnet::ext::CustomOp::setCreateOpState	(	createOpState_t	func,
		const char *	ctx
	)

◆ setForward()

CustomOp& mxnet::ext::CustomOp::setForward	(	fcomp_t	fcomp,
		const char *	ctx
	)

◆ setInferShape()

CustomOp& mxnet::ext::CustomOp::setInferShape ( inferShape_t func )

◆ setInferSType()

CustomOp& mxnet::ext::CustomOp::setInferSType ( inferSType_t func )

◆ setInferType()

CustomOp& mxnet::ext::CustomOp::setInferType ( inferType_t func )

◆ setIsSubgraphOp()

CustomOp& mxnet::ext::CustomOp::setIsSubgraphOp ( )

◆ setMutateInputs()

CustomOp& mxnet::ext::CustomOp::setMutateInputs ( mutateInputs_t func )

◆ setParseAttrs()

CustomOp& mxnet::ext::CustomOp::setParseAttrs ( parseAttrs_t func )

Member Data Documentation

◆ backward_ctx_cstr

std::vector<const char*> mxnet::ext::CustomOp::backward_ctx_cstr

◆ backward_fp

std::vector<fcomp_t> mxnet::ext::CustomOp::backward_fp

◆ create_op_ctx_cstr

std::vector<const char*> mxnet::ext::CustomOp::create_op_ctx_cstr

◆ create_op_fp

std::vector<createOpState_t> mxnet::ext::CustomOp::create_op_fp

◆ forward_ctx_cstr

std::vector<const char*> mxnet::ext::CustomOp::forward_ctx_cstr

vector repr of ctx map to be easily loaded from c_api

◆ forward_fp

std::vector<fcomp_t> mxnet::ext::CustomOp::forward_fp

◆ infer_shape

inferShape_t mxnet::ext::CustomOp::infer_shape

◆ infer_storage_type

inferSType_t mxnet::ext::CustomOp::infer_storage_type

◆ infer_type

inferType_t mxnet::ext::CustomOp::infer_type

◆ isSGop

bool mxnet::ext::CustomOp::isSGop

◆ mutate_inputs

mutateInputs_t mxnet::ext::CustomOp::mutate_inputs

◆ name

const char* mxnet::ext::CustomOp::name

operator name

◆ parse_attrs

parseAttrs_t mxnet::ext::CustomOp::parse_attrs

operator functions

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

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

Public Member Functions

Public Attributes