我有一个 TensorFlow 模型,它被编译为 XLA 以与一些 Graphcore IPU 一起使用。出于调试目的,我试图将 XLA 图形转储到 .dot 文件中,以便在浏览器中对其进行可视化。
为此,我使用以下标志:
--xla_dump_to=/mnt/data/perf_prof_5/xla_dump --xla_dump_hlo_as_dot --xla_dump_hlo_pass_re=forward-allocation --xla_hlo_graph_sharding_color
Run Code Online (Sandbox Code Playgroud)
但是,我得到了多个输出文件,但不确定哪一个是正确的。它们的名称和编号也根据我是否编译图形(使用ipu.compile指令)而有所不同。
哪个文件包含图形以及名称的含义是什么?