Watch talks from the 2019 TensorFlow Dev Summit Watch now

tf.contrib.model_pruning.strip_pruning_vars_fn

tf.contrib.model_pruning.strip_pruning_vars_fn(
    input_graph_def,
    output_node_names
)

Defined in tensorflow/contrib/model_pruning/python/strip_pruning_vars_lib.py.

Removes mask variable from the graph.

Replaces the masked_weight tensor with element-wise multiplication of mask and the corresponding weight variable.

Args:

  • input_graph_def: A GraphDef in which the variables have been converted to constants. This is typically the output of tf.graph_util.convert_variables_to_constant()
  • output_node_names: List of name strings for the result nodes of the graph

Returns:

A GraphDef in which pruning-related variables have been removed