p19647583
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

model input and output check

This commit is contained in:
ling 2021-04-17 15:11:53 +08:00
parent 7951d3f8cb
commit 1ebf0943d0
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/lite/src/lite_model.cc
View File

@ -23,6 +23,7 @@
#include <unordered_map>
#include <memory>
#include "src/common/prim_util.h"
#include "src/common/graph_util.h"
#ifdef ENABLE_V0
#include "src/ops/compat/compat_register.h"
#endif
@ -219,6 +220,12 @@ int LiteModel::SubGraphVerify() const {
auto tensor_size = this->all_tensors_.size();
auto node_size = this->all_nodes_.size();
if (sub_graphs_[0]->input_indices_.size() == 0 || GetGraphInputNodes(this).size() == 0 ||
sub_graphs_[0]->output_indices_.size() == 0 || GetGraphOutputNodes(this).size() == 0) {
MS_LOG(ERROR) << "The model has invalid input and output, please check";
return RET_ERROR;
}
for (auto &graph : this->sub_graphs_) {
if (graph == nullptr) {
MS_LOG(ERROR) << "graph is null.";