tff.learning.models.from_keras_model

tff.learning.models.from_keras_model(
    keras_model: tf.keras.Model,
    loss: Loss,
    input_spec,
    loss_weights: Optional[list[float]] = None,
    metrics: Optional[Union[list[tf.keras.metrics.Metric], list[Callable[[], tf.keras.
        metrics.Metric]]]] = None
) -> tff.learning.models.VariableModel

Used in the notebooks

Used in the tutorials
Federated Learning for Image Classification Federated Learning for Text Generation Differential Privacy in TFF TFF simulations with accelerators