The public API for contributors who develop federated learning algorithms.
Defined in
python/learning/framework/__init__.py.
Classes
class ClientDeltaFn:
Represents a client computation that produces an update to a model.
class ClientOutput: Structure
for outputs returned from clients during federated optimization.
class EnhancedModel: A
wrapper around a Model that adds sanity checking and metadata helpers.
class EnhancedTrainableModel:
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.