提交 95afdb32 编写于 作者: M mindspore-ci-bot 提交者: Gitee

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

Merge pull request !4642 from sunsuodong/topkv2_parser
......@@ -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;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册