def METHOD_NAME(self, model):
    """Freezes the model.
    Args:
      model: Function.
    Returns:
      root: AutoTrackable object with original ConcreteFunction.
      output_func: frozen ConcreteFunction.
    """
    root = tf.Module()
    root.f = model
    input_func = root.f.get_concrete_function()
    output_func = convert_to_constants.convert_variables_to_constants_v2(
        input_func, lower_control_flow=False
    )
    return root, output_func