W3cubDocs

/TensorFlow Python

Module: tf.graph_util

Module tf.graph_util

Helpers to manipulate a tensor graph in python.

Members

convert_variables_to_constants(...): Replaces all the variables in a graph with constants of the same values.

extract_sub_graph(...): Extract the subgraph that can reach any of the nodes in 'dest_nodes'.

must_run_on_cpu(...): Returns True if the given node_def must run on CPU, otherwise False.

remove_training_nodes(...): Prunes out nodes that aren't needed for inference.

tensor_shape_from_node_def_name(...): Convenience function to get a shape from a NodeDef's input string.

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

© 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