Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tff.learning.framework.enhance

View source on GitHub

Wraps a tff.learning.Model as an EnhancedModel.

tff.learning.framework.enhance(
    model
)

Args:

Returns:

An EnhancedModel. If model has already been wrapped as such, this is a no-op.