pytorch
6d97a80d - [fx][graph_drawer] Improve graph drawer coloring and tensor_meta handling (#58699)

Commit
4 years ago
[fx][graph_drawer] Improve graph drawer coloring and tensor_meta handling (#58699) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58699 Make `call_function`/`call_method` random colors based on their target name. This coloring is stable according to the name of the target. Also handle tensor_meta more elegantly for quantized types, including print q_scale/q_zero_point if they're used. Test Plan: Tested locally Reviewed By: chenccfb, 842974287 Differential Revision: D28580333 fbshipit-source-id: ad9961e1106a1bfa5a018d009b0ddb8802d2163c
Author
Parents
Loading