tf.graph_util.extract_sub_graph

tf.graph_util.extract_sub_graph(
    graph_def,
    dest_nodes
)

Defined in tensorflow/python/framework/graph_util_impl.py.

Extract the subgraph that can reach any of the nodes in 'dest_nodes'.

Args:

  • graph_def: A graph_pb2.GraphDef proto.
  • dest_nodes: A list of strings specifying the destination node names.

Returns:

The GraphDef of the sub-graph.

Raises:

  • TypeError: If 'graph_def' is not a graph_pb2.GraphDef proto.