Represents a client computation that produces an update to a model.

variables Returns all the variables of this object.

Note this only includes variables that are part of the state of this object, and not the model variables themselves.



View source

Defines the complete client computation.

Typically implementations should be decorated with tf.function.

dataset A producing batches than can be fed to tff.learning.Model.forward_pass.
initial_weights A dictionary of initial values for all trainable and non-trainable model variables, keyed by name. This will be supplied by the server in Federated Averaging.

An optimizer_utils.ClientOutput namedtuple.