mxnet
|
An abstract class for subgraph property. More...
#include <lib_api.h>
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... | |
An abstract class for subgraph property.
|
inline |
|
inlineexplicit |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
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 |