Browse Source

topk have two output

tags/v0.7.0-beta
sunsuodong 5 years ago
parent
commit
47dbac6803
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      mindspore/lite/tools/converter/parser/tflite/tflite_topk_v2_parser.cc

+ 4
- 2
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, AddOpInput(op, tensors_id, tensors_format, tensors_id_map,
tflite_op->inputs[0], tensors_id->size(), tflite_tensors.size(), schema::Format_NHWC); 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; return RET_OK;
} }




Loading…
Cancel
Save