mxnet
Public Member Functions | Public Attributes | List of all members
CustomPartitioner Class Reference

An abstract class for subgraph property. More...

#include <lib_api.h>

Collaboration diagram for CustomPartitioner:
Collaboration graph

Public Member Functions

 CustomPartitioner ()
 
 CustomPartitioner (const char *backend_name)
 
CustomPartitioneraddStrategy (const char *prop_name, const char *sg_name)
 
CustomPartitionersetSupportedOps (const char *prop_name, supportedOps_t fn)
 
CustomPartitionersetCreateSelector (const char *prop_name, createSelector_t fn)
 
CustomPartitionersetReviewSubgraph (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_tsupported_map
 
std::map< std::string, createSelector_tselector_map
 
std::map< std::string, reviewSubgraph_treview_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::CustomPartitioner ( )
inline
CustomPartitioner::CustomPartitioner ( const char *  backend_name)
inlineexplicit

Member Function Documentation

CustomPartitioner& CustomPartitioner::addStrategy ( const char *  prop_name,
const char *  sg_name 
)
inline
createSelector_t CustomPartitioner::getCreateSelector ( int  stg_id)
inline
reviewSubgraph_t CustomPartitioner::getReviewSubgraph ( int  stg_id)
inline
supportedOps_t CustomPartitioner::getSupportedOps ( int  stg_id)
inline
CustomPartitioner& CustomPartitioner::setCreateSelector ( const char *  prop_name,
createSelector_t  fn 
)
inline
CustomPartitioner& CustomPartitioner::setReviewSubgraph ( const char *  prop_name,
reviewSubgraph_t  fn 
)
inline
CustomPartitioner& CustomPartitioner::setSupportedOps ( const char *  prop_name,
supportedOps_t  fn 
)
inline

Member Data Documentation

const char* CustomPartitioner::name

partitioner name

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

subgraph operator name

std::map<std::string, reviewSubgraph_t> CustomPartitioner::review_map
std::map<std::string, createSelector_t> CustomPartitioner::selector_map
std::vector<const char*> CustomPartitioner::strategies

strategy names

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

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