Save the date! Google I/O returns May 18-20 Register now

tff.learning.framework.weights_type_from_model

Creates a tff.Type from a tff.learning.Model or callable that constructs a model.

tff.learning.framework.weights_type_from_model(
    model: Union[model_lib.Model, Callable[[], model_lib.Model]]
) -> tff.StructType

model a tff.learning.Model instance, or a no-arg callable that returns a model.

A tff.StructType representing the TFF type of the ModelWeights structure for model.