CODAIT / graph_def_editor

GraphDef Editor: A port of the TensorFlow contrib.graph_editor package that operates over serialized graphs
Apache License 2.0
31 stars 16 forks source link

Adding visualization functionality to the graph_def_editor #46

Closed vlasenkoalexey closed 3 years ago

vlasenkoalexey commented 3 years ago

Ability to visualize graphs is especially useful when experimenting with rewrites in Colab/Jupyter Notebook.

Idea is based on: https://github.com/akimach/tfgraphviz

Demo: https://colab.research.google.com/drive/1HZmNR48VsltuksbGAOHWco1VwNTV9u7W#scrollTo=NpGImSh9zIdV