[FX] make visualizer produce different formatted output (#64699)
authorYinghai Lu <yinghai@fb.com>
Thu, 9 Sep 2021 01:20:46 +0000 (18:20 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 9 Sep 2021 01:22:12 +0000 (18:22 -0700)
commit0ef32625a8b44124d8cfadb981ff8ef482c5c1c0
tree2b26d050f79b195e88affa93ba7a5ff95b653cb9
parent86e3b2727e3833ee6008f1f42fc1533d1b5bc1a3
[FX] make visualizer produce different formatted output (#64699)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64699

Previously we just hardcode to svg format. We should give folks a choice in terms of what format they want to see. If we give a weird extension like .abc and this will error out and we expect this to be the right behavior.

Reviewed By: houseroad

Differential Revision: D30718883

fbshipit-source-id: fe8827262f94ea6887999bb225de763d1909eef8
torch/fx/experimental/fx_acc/acc_utils.py