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

!4642 [MS][LITE][Develop]fix topkv2 parser, topkv2 have two output 

Merge pull request !4642 from sunsuodong/topkv2_parser
This commit is contained in:
mindspore-ci-bot 2020-08-18 11:31:45 +08:00 committed by Gitee
parent c22b56e8e0 47dbac6803
commit 95afdb3229
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/lite/tools/converter/parser/tflite/tflite_topk_v2_parser.cc
View File

@ -56,8 +56,10 @@ STATUS TfliteTopKV2Parser::Parse(const std::unique_ptr<tflite::OperatorT> &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;
}