tff.learning.models.weights_type_from_model

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

model A tff.learning.models.VariableModel instance, or a no-arg callable that returns a model.

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