|
|
|
@@ -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; |
|
|
|
} |
|
|
|
|
|
|
|
|