Browse Source

!362 fix centos compile error

Merge pull request !362 from wangzhengjun/centos_compile
pull/361/MERGE
i-robot Gitee 4 years ago
parent
commit
3944fb31b0
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      parser/common/acl_graph_parser_util.cc

+ 1
- 1
parser/common/acl_graph_parser_util.cc View File

@@ -55,7 +55,7 @@ const int kWarningThreshold = 536870912 * 2; // 536870912 represent 512M
const uint32_t kSetOutputWithNodeAndIndex = 0x1; const uint32_t kSetOutputWithNodeAndIndex = 0x1;
const uint32_t kSetOutputWithTensorName = 0x2; const uint32_t kSetOutputWithTensorName = 0x2;
const uint32_t kSetOutputModeMixed = 0x3; const uint32_t kSetOutputModeMixed = 0x3;
const std::unordered_set<domi::FrameworkType> kSupportTensorAsOutput = {
const std::set<domi::FrameworkType> kSupportTensorAsOutput = {
domi::CAFFE, domi::CAFFE,
domi::ONNX domi::ONNX
}; };


Loading…
Cancel
Save