Tensorflow 2 code to dump a SavedModel graph to Tensorboard format as well as text format

dump_graph.py

import tensorflow as tf
import sys

if __name__ == '__main__':
    graph = tf.saved_model.load(sys.argv[1]).signatures['serving_default'].graph
    with graph.as_default():
        writer = tf.compat.v1.summary.FileWriter(sys.argv[2])
        writer.add_graph(graph)
        writer.close()
    tf.io.write_graph(graph.as_graph_def(), sys.argv[2], "graph.pbtxt")