tfgraphviz is a module to visualize a TensorFlow's data flow graph like TensorBoard using Graphviz. tfgraphviz enables to provide a visualization of tensorflow graph on Jupyter Notebook without TensorBoard.
Use pip to install:
$ pip install graphviz
$ pip install tfgraphviz
The only dependency is Graphviz.
macOS:
$ brew install graphviz
Ubuntu:
$ apt-get install graphviz
import tensorflow as tf
import tfgraphviz as tfg
g = tf.Graph()
with g.as_default():
a = tf.constant(1, name="a")
b = tf.constant(2, name="b")
c = tf.add(a, b, name="add")
tfg.board(g)
This package is distributed under the MIT license.