tf.contrib.model_pruning.graph_def_from_checkpoint(
checkpoint_dir,
output_node_names
)
Defined in tensorflow/contrib/model_pruning/python/strip_pruning_vars_lib.py
.
Converts checkpoint data to GraphDef.
Reads the latest checkpoint data and produces a GraphDef in which the variables have been converted to constants.
Args:
checkpoint_dir
: Path to the checkpoints.output_node_names
: List of name strings for the result nodes of the graph.
Returns:
A GraphDef from the latest checkpoint
Raises:
ValueError
: if no checkpoint is found