Browse Source

Description:Support model_exit in GE

Team:HISI_SW
Feature or Bugfix:Feature
pull/203/head
l00444296 4 years ago
parent
commit
1fae4cde59
1 changed files with 23 additions and 0 deletions
  1. +23
    -0
      inc/external/ge/ge_api_types.h

+ 23
- 0
inc/external/ge/ge_api_types.h View File

@@ -276,6 +276,11 @@ namespace ir_option {
static const char *const INPUT_FORMAT = "input_format"; static const char *const INPUT_FORMAT = "input_format";
static const char *const INPUT_SHAPE = "input_shape"; static const char *const INPUT_SHAPE = "input_shape";
static const char *const OP_NAME_MAP = "op_name_map"; static const char *const OP_NAME_MAP = "op_name_map";
static const char *const IS_DYNAMIC_INPUT = "is_dynamic_input";
static const char *const IS_INPUT_ADJUST_HW_LAYOUT = "is_input_adjust_hw_layout";
static const char *const IS_OUTPUT_ADJUST_HW_LAYOUT = "is_output_adjust_hw_layout";
static const char *const ENABLE_SCOPE_FUSION_PASSES = "enable_scope_fusion_passes";
static const char *const OUTPUT = "output";
static const char *const DYNAMIC_BATCH_SIZE = kDynamicBatchSize; static const char *const DYNAMIC_BATCH_SIZE = kDynamicBatchSize;
static const char *const DYNAMIC_IMAGE_SIZE = kDynamicImageSize; static const char *const DYNAMIC_IMAGE_SIZE = kDynamicImageSize;
static const char *const DYNAMIC_DIMS = kDynamicDims; static const char *const DYNAMIC_DIMS = kDynamicDims;
@@ -316,6 +321,24 @@ namespace ir_option {
INPUT_FP16_NODES, INPUT_FP16_NODES,
LOG_LEVEL LOG_LEVEL
}; };

// for interface: aclgrphParse
const std::set<std::string> ir_Parser_suppported_options = {
INPUT_FORMAT,
INPUT_SHAPE,
OP_NAME_MAP,
IS_DYNAMIC_INPUT,
INPUT_FP16_NODES,
IS_INPUT_ADJUST_HW_LAYOUT,
IS_OUTPUT_ADJUST_HW_LAYOUT,
OUTPUT,
OUTPUT_TYPE,
OUT_NODES,
COMPRESS_WEIGHT_CONF,
ENABLE_SCOPE_FUSION_PASSES,
LOG_LEVEL
};

// for interface: aclgrphBuildInitialize // for interface: aclgrphBuildInitialize
const std::set<std::string> global_options = { const std::set<std::string> global_options = {
CORE_TYPE, CORE_TYPE,


Loading…
Cancel
Save