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

Handle ONNX topography structure #731

Closed IDownToEarth closed 10 months ago

IDownToEarth commented 11 months ago

Hi Just a quick question about the ONNX parsing. According to my understanding, if we do inference, we will need to know the topography of the net, so we will have a module to handle the topography. I find the file IOnnxParser.hpp, but I don't find the detailed code to handle the node info and node order in this file. So I just want to know does there exist such a module to handle the topography structure of ONNX. Thanks a lot.

MikeJKelly commented 11 months ago

Hi @IDownToEarth I think the code you want is in armnn/src/armnnOnnxParser/OnnxParser.cpp . Our support for ONNX operations is quite limited at the present but you may find what you want there.

Best regards, Mike

KevinARM commented 11 months ago

Hi @IDownToEarth

There is also a list of supported layers here: https://arm-software.github.io/armnn/23.05/parsers.xhtml#S5_onnx_parser

Regards, Kevin

John-ARM commented 10 months ago

Closed due to inactivity, if you need any more information please reopen the issue or create a new one. Thank you, John