From 47dbac6803341cb5e162c2bebf221a6f7b26729a Mon Sep 17 00:00:00 2001 From: sunsuodong Date: Tue, 18 Aug 2020 10:13:46 +0800 Subject: [PATCH] topk have two output --- .../tools/converter/parser/tflite/tflite_topk_v2_parser.cc | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore/lite/tools/converter/parser/tflite/tflite_topk_v2_parser.cc b/mindspore/lite/tools/converter/parser/tflite/tflite_topk_v2_parser.cc index 7c8fbb0b1e..fc90e1ead8 100644 --- a/mindspore/lite/tools/converter/parser/tflite/tflite_topk_v2_parser.cc +++ b/mindspore/lite/tools/converter/parser/tflite/tflite_topk_v2_parser.cc @@ -56,8 +56,10 @@ STATUS TfliteTopKV2Parser::Parse(const std::unique_ptr &tflit AddOpInput(op, tensors_id, tensors_format, tensors_id_map, tflite_op->inputs[0], tensors_id->size(), tflite_tensors.size(), schema::Format_NHWC); - AddOpOutput(op, tensors_id, tensors_format, tensors_id_map, - tflite_op->outputs[0], tensors_id->size(), tflite_tensors.size(), schema::Format_NHWC); + for (int i = 0; i < tflite_op->outputs.size(); i++) { + AddOpOutput(op, tensors_id, tensors_format, tensors_id_map, + tflite_op->outputs[i], tensors_id->size(), tflite_tensors.size(), schema::Format_NHWC); + } return RET_OK; }