Defined in
learning/framework/__init__.py.
The public API for contributors who develop federated learning algorithms.
Classes
class ClientDeltaFn:
Represents a client computation that produces an update to a model.
class ClientOutput:
ClientOutput(weights_delta, weights_delta_weight, model_output,
optimizer_output)
class EnhancedModel: A
wrapper around a Model that adds sanity checking and metadata helpers.
class ModelWeights: A
container for the trainable and non-trainable variables of a Model.
Functions
build_model_delta_optimizer_process(...):
Constructs
tff.utils.IterativeProcess
for Federated Averaging or SGD.