Browse Source

Pre Merge pull request !1049 from 赵之轩/my_dev

pull/1049/MERGE
赵之轩 Gitee 5 years ago
parent
commit
f747fbe66b
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/single_op/single_op_model.cc

+ 2
- 2
ge/single_op/single_op_model.cc View File

@@ -43,8 +43,7 @@ using std::vector;
namespace ge {
namespace {
const size_t kDataOutputNum = 1;
} // namespace
static Status IfInferDepend(GeModelPtr &ge_model, bool &flag) {
Status IfInferDepend(GeModelPtr &ge_model, bool &flag) {
auto comp_graph = GraphUtils::GetComputeGraph(ge_model->GetGraph());
for (const auto &node : comp_graph->GetAllNodes()) {
auto op_desc = node->GetOpDesc();
@@ -57,6 +56,7 @@ static Status IfInferDepend(GeModelPtr &ge_model, bool &flag) {
}
return SUCCESS;
}
} // namespace

SingleOpModel::SingleOpModel(const std::string &model_name, const void *model_data, uint32_t model_size)
: model_name_(model_name), ori_model_data_(model_data), ori_model_size_(model_size) {}


Loading…
Cancel
Save