How to graph tf.keras model in Tensorflow-2.0? : <p>I upgraded to Tensorflow 2.0 and there is no <code>tf.summary.FileWriter("tf_graphs", sess.graph)</code>. I was looking through some other StackOverflow questions on this and they said to use <code>tf.compat.v1.summary etc</code>. Surely there must be a way to graph and visualize a tf.keras model in Tensorflow version 2. What is it? I'm looking for a tensorboard output like the one below. Thank you!</p>

<p><a href="https://i.stack.imgur.com/Zhc3g.png" rel="noreferrer"><img src="https://i.stack.imgur.com/Zhc3g.png" alt="enter image description here"></a></p>
