Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfl.model_info.ModelGraph

View source on GitHub

Model info and parameter as a graph.

@staticmethod
tfl.model_info.ModelGraph(
    _cls, nodes, output_node
)

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

Attributes:

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