sbjkx
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6477 [MS][LITE][Develop]input index check 

Merge pull request !6477 from chenjianping/lite_dev
This commit is contained in:
mindspore-ci-bot 2020-09-18 16:03:48 +08:00 committed by Gitee
parent 31ad1654a1 b3b4fc4e6b
commit a9af90a3dd
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
mindspore/lite/internal/src/lite_session.cc
View File

@ -105,6 +105,10 @@ int LiteSession::CompileGraph(Model *model) {
InitFuncs();
g_model = model;
for (auto in : g_model->input_indices_) {
if (in >= g_model->all_tensors_.size() || in < 0) {
LITE_LOG_ERROR("Invalid input indices!");
return RET_PARAM_INVALID;
}
g_model->all_tensors_[in]->data_ = g_allocator.Malloc(g_model->all_tensors_[in]->Size());
}
g_infershape_interrupt = false;
@ -118,7 +122,12 @@ int LiteSession::CompileGraph(Model *model) {
TensorPtrVector LiteSession::GetInputs() const {
TensorPtrVector in(g_model->input_indices_.size());
for (size_t i = 0; i < g_model->input_indices_.size(); ++i) {
in.at(i) = g_model->all_tensors_[g_model->input_indices_[i]];
auto index = g_model->input_indices_[i];
if (index < 0 || index >= g_model->all_tensors_.size()) {
LITE_ERROR_LOG("Invalid input index: %u", index);
return TensorPtrVector();
}
in.at(i) = g_model->all_tensors_[index];
}
return in;
}
@ -130,7 +139,12 @@ TensorPtrVector LiteSession::GetOutputsByNodeName(const String &node_name) const
TensorPtrVector LiteSession::GetOutputs() const {
TensorPtrVector out(g_model->output_indices_.size());
for (size_t i = 0; i < g_model->output_indices_.size(); ++i) {
out.at(i) = g_model->all_tensors_[g_model->output_indices_[i]];
auto index = g_model->output_indices_[i];
if (index < 0 || index >= g_model->all_tensors_.size()) {
LITE_ERROR_LOG("Invalid output index: %u", index);
return TensorPtrVector();
}
out.at(i) = g_model->all_tensors_[index];
}
return out;
}