|
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 |
1.8.11