Multiple enhancements to graphviz output:

  - nodes organized by subgraphs inferred from array names
  - subgraphs and ops include math op count estimates
  - better organized labels on all nodes
  - operators have numbered inputs and outputs when two or more exist
  - specific node shapes for inputs/outputs
  - node color scheme unchanged

PiperOrigin-RevId: 233446311