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

!8832 [MSLITE]add two multi-input models to the entrance guard 

From: @probiotics_53
Reviewed-by: @hangangqiang
Signed-off-by: @hangangqiang
This commit is contained in:
mindspore-ci-bot 2020-11-22 09:40:11 +08:00 committed by Gitee
parent 4569a4695e 71abd217c7
commit 5e3b135130
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/lite_session.cc
View File

@ -197,9 +197,9 @@ void LiteSession::InitGraphInputMap(const lite::Model *model) {
for (auto in_node_index : graph_input_node_indexes) {
auto in_node = model->all_nodes_[in_node_index];
MS_ASSERT(in_node != nullptr);
MS_ASSERT(this->input_map_.find(in_node->name_) == this->input_map_.end());
auto in_size = in_node->input_indices_.size();
for (size_t i = 0; i < in_size; ++i) {
MS_ASSERT(this->input_map_.find(in_node->name_) + std::to_string(i) == this->input_map_.end());
auto in_tensor_index = size_t(in_node->input_indices_[i]);
bool is_graph_input = false;
for (size_t j = 0; j < graph_in_size; ++j) {

2
mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg
View File

@ -1,5 +1,6 @@
lite-model_arbitrary-image-stylization-inceptionv3_dr_transfer_1.tflite
lite-model_arbitrary-image-stylization-inceptionv3_int8_transfer_1.tflite
lite-model_arbitrary-image-stylization-inceptionv3_fp16_transfer_1.tflite
lite-model_arbitrary-image-stylization-inceptionv3-dynamic-shapes_dr_transfer_1.tflite
lite-model_cartoongan_dr_1.tflite
mindspore_efficientnet_b0.mindir
@ -25,3 +26,4 @@ efficientnet_lite4_int8_2.tflite
mtk_transformer_encoder.tflite
mtk_transformer_decoder_joint.tflite
ml_ei_facedetection.onnx
mobilebert_1_default_1.tflite