Browse Source

display model info

pull/647/head
wangwenhua1@huawei.com 4 years ago
parent
commit
366556bd1f
2 changed files with 6 additions and 6 deletions
  1. +1
    -1
      inc/framework/omg/omg.h
  2. +5
    -5
      inc/framework/omg/omg_inner_types.h

+ 1
- 1
inc/framework/omg/omg.h View File

@@ -21,8 +21,8 @@
#include <string>
#include <unordered_map>
#include <vector>
#include "framework/omg/omg_inner_types.h"
#include "framework/common/ge_types.h"
#include "framework/omg/omg_inner_types.h"
#include "framework/omg/parser/parser_inner_ctx.h"
#include "proto/ge_ir.pb.h"
#include "proto/om.pb.h"


+ 5
- 5
inc/framework/omg/omg_inner_types.h View File

@@ -43,11 +43,11 @@ namespace ge {
* @brief run model
*/
enum RunMode {
GEN_OM_MODEL = 0, // generate offline model file
MODEL_TO_JSON = 1, // convert to JSON file
ONLY_PRE_CHECK = 3, // only for pre-check
PBTXT_TO_JSON = 5, // pbtxt to json
DISPLAY_OM_INFO = 6 // display model info
GEN_OM_MODEL = 0, // generate offline model file
MODEL_TO_JSON = 1, // convert to JSON file
ONLY_PRE_CHECK = 3, // only for pre-check
PBTXT_TO_JSON = 5, // pbtxt to json
DISPLAY_OM_INFO = 6 // display model info
};

///


Loading…
Cancel
Save