search

ARM-software / armnn

Arm NN ML Software. The code here is a read-only mirror of https://review.mlplatform.org/admin/repos/ml/armnn
https://developer.arm.com/products/processors/machine-learning/arm-nn
MIT License
1.15k stars 307 forks source link

Fix TfLiteParserImpl::GetNetworkOutputBindingInfo #602

Closed steven9046 closed 2 years ago

steven9046 commented 2 years ago

In the previous version GetNetworkOutputBindingInfo will use m_OverridenOutputShapes to get the output tensor shape. The order is got from GetSubgraphOutputs(). But this order may be different from m_OverridenOutputShapes wich is got from custom operator 'TFLite_Detection_PostProcess'. Map the outputs shape by layer name can solve this problem. Look this issue #585 .

TeresaARM commented 2 years ago

Hi @steven9046

Thank you very much for your contribution to ArmNN, we would really love to review it and merge it.

However, instead of a pull request, we expect a commit in Gerrit. You should create a Gerrit review account at mlplatform.org and make your contribution there. You can use your GitHub credentials when creating your account.

The process for contributing to ArmNN is outlined in this Contributor Guide.

Please let us know if you find any issue.

Thank you very much. Kindest regards

TeresaARM commented 2 years ago

Output bindings fixed in 22.02