Browse Source

Pre Merge pull request !201 from lixiwen1/development

pull/201/MERGE
lixiwen1 Gitee 4 years ago
parent
commit
64856aef3d
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      inc/framework/omg/parser/parser_inner_ctx.h

+ 1
- 0
inc/framework/omg/parser/parser_inner_ctx.h View File

@@ -31,6 +31,7 @@ namespace ge {
struct ParserContext {
// format of the input specified by the command line
std::unordered_map<std::string, domiTensorFormat_t> input_nodes_format_map;
std::vector<domiTensorFormat_t> output_formats;
// user-designate input dims
std::vector<std::pair<std::string, std::vector<int64_t>>> user_input_dims;
std::unordered_map<std::string, std::vector<int64_t>> input_dims;


Loading…
Cancel
Save