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.14k stars 307 forks source link

How to close optimization of graph or save optimized graph in dot/tflite format when I run test/executeNetwork and set --tflite-executor as delegate? Does armnn provide options? #745

Closed hgy20 closed 2 months ago

tracyn-arm commented 7 months ago

Hi. You can save a dot file using the "-v, --visualize-optimized-model" parameter of ExecuteNetwork. The saving of the dot file happens after the network is optimized and I think it is placed in the same directory of the model. This is done in the armnn_delegate.cpp file in the classic and opaque code paths after armnn::Optimize(...) is called. Please let us know if you are seeing something unexpected. Thanks.

hgy20 commented 6 months ago

Hi @tracyn-arm, I use tflite-executor as delegate bur it seens that option "-v" only works when i use parser as delegate. Thanks.

tracyn-arm commented 6 months ago

Thank-you. I am seeing the problem and looking for a fix.

tracyn-arm commented 6 months ago

A fix was returned: https://review.mlplatform.org/c/ml/armnn/+/10867. Thank-you for reporting this issue.