tfl.model_info.ModelGraph

View source on GitHub

Model info and parameter as a graph.

Note that this is not a TF graph, but rather a graph of python object that describe model structure and parameters.

nodes List of all the nodes in the model.
output_node The output node of the model.