Google I/O returns May 18-20! Reserve space and build your schedule Register now


Represents the state of the server carried between rounds.

model a ModelWeights structure, containing Tensors or Variables.
optimizer_state a list of Tensors or Variables, in the order returned by optimizer.variables()
delta_aggregate_state state (possibly empty) of the delta_aggregate_fn.
model_broadcast_state state (possibly empty) of the model_broadcast_fn.