New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

mlir2onnx tool optimize (for visualizing *_final.mlir files) #80

Open

charlesxzb opened this issue Feb 18, 2023 · 0 comments

Labels

Bonus enhancement

Collaborator

charlesxzb commented Feb 18, 2023 •

edited

Optimize mlir2onnx.py so that *_final.mlir can be converted to .pb file and visualized in Netron.

The main difference between *.mlir and *_final.mlir is the latter contains information of the subnet and layer group.

Source Code Path:
tpu-mlir/python/tools/mlir2onnx.py

To test:

$ cd tpu-mlir/regression  
$ ./run_model.sh resnet50_tf  
$ mlir2onnx.py -m regression_out/resnet50_tf_bm1684x/resnet50_tf_bm1684x_int8_asym_final.mlir -o resnet50_tf_final.pb

Bonus: 500pt

The text was updated successfully, but these errors were encountered:

charlesxzb added enhancement Bonus labels

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment