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.
mxnet::ext::CustomPartitioner::CustomPartitioner | ( | ) |
|
explicit |
CustomPartitioner& mxnet::ext::CustomPartitioner::addStrategy | ( | const char * | prop_name, |
const char * | sg_name | ||
) |
createSelector_t mxnet::ext::CustomPartitioner::getCreateSelector | ( | int | stg_id | ) |
reviewSubgraph_t mxnet::ext::CustomPartitioner::getReviewSubgraph | ( | int | stg_id | ) |
supportedOps_t mxnet::ext::CustomPartitioner::getSupportedOps | ( | int | stg_id | ) |
CustomPartitioner& mxnet::ext::CustomPartitioner::setCreateSelector | ( | const char * | prop_name, |
createSelector_t | fn | ||
) |
CustomPartitioner& mxnet::ext::CustomPartitioner::setReviewSubgraph | ( | const char * | prop_name, |
reviewSubgraph_t | fn | ||
) |
CustomPartitioner& mxnet::ext::CustomPartitioner::setSupportedOps | ( | const char * | prop_name, |
supportedOps_t | fn | ||
) |
const char* mxnet::ext::CustomPartitioner::name |
partitioner name
std::vector<const char*> mxnet::ext::CustomPartitioner::op_names |
subgraph operator name
std::map<std::string, reviewSubgraph_t> mxnet::ext::CustomPartitioner::review_map |
std::map<std::string, createSelector_t> mxnet::ext::CustomPartitioner::selector_map |
std::vector<const char*> mxnet::ext::CustomPartitioner::strategies |
strategy names
std::map<std::string, supportedOps_t> mxnet::ext::CustomPartitioner::supported_map |