tf.graph_util.extract_sub_graph(graph_def, dest_nodes)

tf.graph_util.extract_sub_graph(graph_def, dest_nodes)

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.

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