Module: tff.utils

Defined in core/utils/

Utility classes/functions built on top of TensorFlow Federated Core API.

All components that depend on utils should import symbols from this file rather than directly importing individual modules. For this reason, the visibility for the latter is set to private and should remain such. The code in utils must not depend on implementation classes. It should be written against the Core API.


class IterativeProcess: A process that includes an initialization and iterated computation.

class StatefulAggregateFn: A simple container for a stateful aggregation function.

class StatefulBroadcastFn: A simple container for a stateful broadcast function.


assign(...): Creates an op that assigns target from source.

get_variables(...): Creates a set of variables that matches the given type_spec.

identity(...): Applies tf.identity pointwise to source.

update_state(...): Returns a new state (a namedtuple) with updated with kwargs.