f METHOD_NAME(self, model_and_input,
                                      distribution_for_saving,
                                      distribution_for_restoring,
                                      save_in_scope,
                                      experimental_run_tf_function):