tff.learning.framework.parameter_count_from_model

Computes count of trainable parameters for a model.