W3cubDocs

/TensorFlow Python

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.

© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/graph_util/extract_sub_graph