Module: tf.graph_util

Module: tf.graph_util

Module tf.graph_util

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

Helpers to manipulate a tensor graph in python.

Functions

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.