TensorFlow 2.0 Beta is available Learn more

Module: tff.utils

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

Defined in python/core/utils/__init__.py.

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.

federated_max(...): Aggregation to find the maximum value from the tff.CLIENTS.

federated_min(...): Aggregation to find the minimum value from the tff.CLIENTS.

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 with new values for fields in kwargs.