tff.learning.framework.weights_type_from_model

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

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.