open source engine_layer_visualize.py (#66301)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/66301
Test Plan: testinprod
Reviewed By: 842974287
Differential Revision: D31495086
fbshipit-source-id: 1f5505d6baac66eca11a35ce9532d6c7c7513190