Browse Source

add option for ir

tags/v1.2.0
wxl 3 years ago
parent
commit
0d85b6c0a8
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      inc/external/ge/ge_api_types.h

+ 4
- 1
inc/external/ge/ge_api_types.h View File

@@ -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,


Loading…
Cancel
Save