From 71abd217c7cef733a997c0d98b7a06c8d7952753 Mon Sep 17 00:00:00 2001 From: zengxianglong Date: Sat, 21 Nov 2020 15:26:14 +0800 Subject: [PATCH] add two multi-input models --- mindspore/lite/src/lite_session.cc | 2 +- .../lite/test/models_with_several_inputs_or_without_outputs.cfg | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/src/lite_session.cc b/mindspore/lite/src/lite_session.cc index c58c64c722a..7f5e43d2c61 100644 --- a/mindspore/lite/src/lite_session.cc +++ b/mindspore/lite/src/lite_session.cc @@ -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) { diff --git a/mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg b/mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg index 18c9dc96eb0..1efff5e30ca 100644 --- a/mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg +++ b/mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg @@ -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