tfm.core.train_utils.convert_variables_to_constants_v2_as_graph

Replaces all the variables in a graph with constants of the same values.

tfm.core.train_utils.convert_variables_to_constants_v2_as_graph(
    func, lower_control_flow=True, aggressive_inlining=False
)

This function works as same as convert_variables_to_constants_v2, but it returns the intermediate GraphDef as well. This GraphDef contains all the debug information after all the transformations in the frozen phase.

func ConcreteFunction.
lower_control_flow Boolean indicating whether or not to lower control flow ops such as If and While. (default True)
aggressive_inlining Boolean indicating whether or not to do aggressive function inlining (might be unsafe if function has stateful ops, not properly connected to control outputs).

ConcreteFunction containing a simplified version of the original, and also the intermediate GraphDef containing the node debug information for the transformations in the frozen phase.