We currently draw tensors as boxes and operators as circles.  The graph will be complex if large model. We draw the tensor information on the edges and keep only operators as nodes.