A data scientist is working on a machine learning model using TensorFlow and wishes to document the model's architecture alongside the experiment. Which MLflow functionality should they use to log the TensorFlow model graph?