tf.graph_util
Helpers to manipulate a tensor graph in python.
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