|
@@ -293,6 +293,7 @@ const std::string MDL_BANK_PATH_FLAG = "ge.mdl_bank_path"; |
|
|
|
|
|
|
|
|
// Configure op bank path |
|
|
// Configure op bank path |
|
|
const std::string OP_BANK_PATH_FLAG = "ge.op_bank_path"; |
|
|
const std::string OP_BANK_PATH_FLAG = "ge.op_bank_path"; |
|
|
|
|
|
const std::string OP_BANK_UPDATE_FLAG = "ge.op_bank_update"; |
|
|
|
|
|
|
|
|
// Graph run mode |
|
|
// Graph run mode |
|
|
enum GraphRunMode { PREDICTION = 0, TRAIN }; |
|
|
enum GraphRunMode { PREDICTION = 0, TRAIN }; |
|
@@ -366,6 +367,7 @@ static const char *const OP_COMPILER_CACHE_DIR = ge::OP_COMPILER_CACHE_DIR; |
|
|
static const char *const OP_COMPILER_CACHE_MODE = ge::OP_COMPILER_CACHE_MODE; |
|
|
static const char *const OP_COMPILER_CACHE_MODE = ge::OP_COMPILER_CACHE_MODE; |
|
|
static const char *const MDL_BANK_PATH = ge::MDL_BANK_PATH_FLAG.c_str(); |
|
|
static const char *const MDL_BANK_PATH = ge::MDL_BANK_PATH_FLAG.c_str(); |
|
|
static const char *const OP_BANK_PATH = ge::OP_BANK_PATH_FLAG.c_str(); |
|
|
static const char *const OP_BANK_PATH = ge::OP_BANK_PATH_FLAG.c_str(); |
|
|
|
|
|
static const char *const OP_BANK_UPDATE = ge::OP_BANK_UPDATE_FLAG.c_str(); |
|
|
static const char *const OP_DEBUG_LEVEL = ge::OP_DEBUG_LEVEL.c_str(); |
|
|
static const char *const OP_DEBUG_LEVEL = ge::OP_DEBUG_LEVEL.c_str(); |
|
|
|
|
|
|
|
|
// for interface: aclgrphBuildModel |
|
|
// for interface: aclgrphBuildModel |
|
@@ -389,7 +391,8 @@ const std::set<std::string> ir_builder_suppported_options = {INPUT_FORMAT, |
|
|
OP_COMPILER_CACHE_DIR, |
|
|
OP_COMPILER_CACHE_DIR, |
|
|
OP_COMPILER_CACHE_MODE, |
|
|
OP_COMPILER_CACHE_MODE, |
|
|
MDL_BANK_PATH, |
|
|
MDL_BANK_PATH, |
|
|
OP_BANK_PATH}; |
|
|
|
|
|
|
|
|
OP_BANK_PATH, |
|
|
|
|
|
OP_BANK_UPDATE}; |
|
|
|
|
|
|
|
|
// for interface: aclgrphParse |
|
|
// for interface: aclgrphParse |
|
|
const std::set<std::string> ir_parser_suppported_options = {INPUT_FORMAT, |
|
|
const std::set<std::string> ir_parser_suppported_options = {INPUT_FORMAT, |
|
|