pull/646/head
wxl 4 years ago
parent 0dfbbe1c1e
commit feb37b2ff3

@ -293,6 +293,8 @@ const std::string MDL_BANK_PATH_FLAG = "ge.mdl_bank_path";
// Configure 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
enum GraphRunMode { PREDICTION = 0, TRAIN };
@ -366,6 +368,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 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_UPDATE = ge::OP_BANK_UPDATE_FLAG.c_str();
static const char *const OP_DEBUG_LEVEL = ge::OP_DEBUG_LEVEL.c_str();
// for interface: aclgrphBuildModel
@ -389,7 +392,8 @@ const std::set<std::string> ir_builder_suppported_options = {INPUT_FORMAT,
OP_COMPILER_CACHE_DIR,
OP_COMPILER_CACHE_MODE,
MDL_BANK_PATH,
OP_BANK_PATH};
OP_BANK_PATH,
OP_BANK_UPDATE};
// for interface: aclgrphParse
const std::set<std::string> ir_parser_suppported_options = {INPUT_FORMAT,

Loading…
Cancel
Save