mxnet::ext::CustomPartitioner Class Reference

An abstract class for subgraph property. More...

#include <lib_api.h>

Collaboration diagram for mxnet::ext::CustomPartitioner:

Public Member Functions
	CustomPartitioner ()
	CustomPartitioner (const char *backend_name)
CustomPartitioner &	addStrategy (const char *prop_name, const char *sg_name)
CustomPartitioner &	setSupportedOps (const char *prop_name, supportedOps_t fn)
CustomPartitioner &	setCreateSelector (const char *prop_name, createSelector_t fn)
CustomPartitioner &	setReviewSubgraph (const char *prop_name, reviewSubgraph_t fn)
supportedOps_t	getSupportedOps (int stg_id)
createSelector_t	getCreateSelector (int stg_id)
reviewSubgraph_t	getReviewSubgraph (int stg_id)

Public Attributes
const char *	name
	partitioner name More...
std::map< std::string, supportedOps_t >	supported_map
std::map< std::string, createSelector_t >	selector_map
std::map< std::string, reviewSubgraph_t >	review_map
std::vector< const char * >	strategies
	strategy names More...
std::vector< const char * >	op_names
	subgraph operator name More...

Detailed Description

An abstract class for subgraph property.

Constructor & Destructor Documentation

CustomPartitioner() [1/2]

mxnet::ext::CustomPartitioner::CustomPartitioner

(

)

CustomPartitioner() [2/2]

mxnet::ext::CustomPartitioner::CustomPartitioner ( const char *  backend_name )

explicit

Member Function Documentation

addStrategy()

CustomPartitioner& mxnet::ext::CustomPartitioner::addStrategy	(	const char *	prop_name,
		const char *	sg_name
	)

getCreateSelector()

createSelector_t mxnet::ext::CustomPartitioner::getCreateSelector

(

int

stg_id

)

getReviewSubgraph()

reviewSubgraph_t mxnet::ext::CustomPartitioner::getReviewSubgraph

(

int

stg_id

)

getSupportedOps()

supportedOps_t mxnet::ext::CustomPartitioner::getSupportedOps

(

int

stg_id

)

setCreateSelector()

CustomPartitioner& mxnet::ext::CustomPartitioner::setCreateSelector	(	const char *	prop_name,
		createSelector_t	fn
	)

setReviewSubgraph()

CustomPartitioner& mxnet::ext::CustomPartitioner::setReviewSubgraph	(	const char *	prop_name,
		reviewSubgraph_t	fn
	)

setSupportedOps()

CustomPartitioner& mxnet::ext::CustomPartitioner::setSupportedOps	(	const char *	prop_name,
		supportedOps_t	fn
	)

Member Data Documentation

name

const char* mxnet::ext::CustomPartitioner::name

partitioner name

op_names

std::vector<const char*> mxnet::ext::CustomPartitioner::op_names

subgraph operator name

review_map

std::map<std::string, reviewSubgraph_t> mxnet::ext::CustomPartitioner::review_map

selector_map

std::map<std::string, createSelector_t> mxnet::ext::CustomPartitioner::selector_map

strategies

std::vector<const char*> mxnet::ext::CustomPartitioner::strategies

strategy names

supported_map

std::map<std::string, supportedOps_t> mxnet::ext::CustomPartitioner::supported_map

---

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

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