|
|
|
@@ -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) {} |
|
|
|
|