Analysis of the resnet 50 model in keras

Hi, I’m currently trying out the resnet 50 model in keras which uses relay IR. I would like to know is there any way to visualize the nodes and the time taken by each node to execute and so on. If there exists one such kindly let me know. If not I would like to build one such if needed.

Best, Vishwas

Is this what you’re looking for?

Hi, Thanks for the response. Yes something similar. But it is not there any more in the repository right? I pulled the code from that link but unable to find tedd.py and test_tedd.py under python/tvm/contrib

If i know which files to go thorugh that would be of great help.

Cheers, Vishwas

This is a PR being reviewed so it’s not in the upstream yet. You could help review it or ask the author if there’s anything you could help.

If you just want to give it a shot, you can checkout that PR branch so that you will be able to see the files.