View source on GitHub |
Model info and parameter as a graph.
tfl.model_info.ModelGraph(
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. |