Visualizing graphs in Tensorboard