There is a an usage of the tool at:

  

      github.com
  

  
    tensorflow/tensorflow/blob/master/tensorflow/compiler/xla/tools/interactive_graphviz.cc#L98:L103



      
      
          
          
If you pass --platform, this tool will compile the HloModule for the given
          platform.  This means that if you acquired your proto from a binary running at a
          particular CL, the HLO graph it ran isn't necessarily the same as the one shown
          here, unless this program was built at the same CL (and our compiler is
          deterministic :).
          
          
Be patient when starting this program if you give it a large input; it has to
          compile the whole thing.
          
          
Usage:
          
          
  interactive_graphviz -- \
              --{hlo_snapshot,hlo_proto,hlo_text}=path/to/binary_proto
              --platform={CUDA,CPU,...}
          )";
          
          
// Unless an explicit width is specified, we will render a neighborhood of
          // kDefaultWidth nodes around the requested instruction.
          constexpr int64_t kDefaultWidth = 2;