tf.graph_util.extract_sub_graph
tf.graph_util.extract_sub_graph
tf.graph_util.extract_sub_graph
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.
© 2017 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