Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tff.utils

View source

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.

build_encoded_broadcast(...): Builds StatefulBroadcastFn for values, to be encoded by encoders.

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

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

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

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

update_state(...): Returns a new state with new values for fields in kwargs.