tf.graph_util.convert_variables_to_constants(sess, input_graph_def, output_node_names, variable_names_whitelist=None, variable_names_blacklist=None)
Replaces all the variables in a graph with constants of the same values.
If you have a trained graph containing Variable ops, it can be convenient to convert them all to Const ops holding the same values. This makes it possible to describe the network fully with a single GraphDef file, and allows the removal of a lot of ops related to loading and saving the variables.
sess
: Active TensorFlow session containing the variables.input_graph_def
: GraphDef object holding the network.output_node_names
: List of name strings for the result nodes of the graph.variable_names_whitelist
: The set of variable names to convert (by default, all variables are converted).variable_names_blacklist
: The set of variable names to omit converting to constants.GraphDef containing a simplified version of the original.
Defined in tensorflow/python/framework/graph_util_impl.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/convert_variables_to_constants