tff.framework.FederatedResolvingStrategy

A strategy for resolving federated types and intrinsics.

Inherits From: FederatingStrategy

tff.framework.FederatedResolvingStrategy(
    executor: tff.framework.FederatingExecutor,
    target_executors: Dict[str, tff.framework.Executor]
)

This strategy implements the federating_executor.FederatingStrategy interface and provides logic for resolving federated types and federated intrinsics on an underlying collection of target executor(s) associated with individual placements.

This strategy only supports the placements:

Note that this strategy does not have a built-in concept of intermediate aggregation, partitioning placements, clustering clients, etc.

executor A federating_executor.FederatingExecutor to use to handle unplaced types, computations, and processing.
target_executors A dict mapping placements to a collection of target executors associated with individual placements. The keys in this dictionary are placement literals. The values can be either single executors (if there only is a single participant associated with that placement, e.g. tff.SERVER) or lists of executors.

TypeError If target_executors is not a dict, where each key is a placement_literals.PlacementLiteral and each value is either an executor_base.Executor or a list of executor_base.Executors.
ValueError If target_executors contains a placement_literals.PlacementLiteral key that is not a kind supported by the FederatedResolvingStrategy.

Methods

close

View source

close()

Release resources associated with this strategy, if any.

compute_federated_aggregate

View source

compute_federated_aggregate(
    *fn_args, **fn_kwargs
)

compute_federated_apply

View source

compute_federated_apply(
    *fn_args, **fn_kwargs
)

compute_federated_broadcast

View source

compute_federated_broadcast(
    *fn_args, **fn_kwargs
)

compute_federated_collect

View source

compute_federated_collect(
    *fn_args, **fn_kwargs
)

compute_federated_eval_at_clients

View source

compute_federated_eval_at_clients(
    *fn_args, **fn_kwargs
)

compute_federated_eval_at_server

View source

compute_federated_eval_at_server(
    *fn_args, **fn_kwargs
)

compute_federated_intrinsic

View source

compute_federated_intrinsic(
    uri, arg=None
)

Returns an embedded call for a federated intrinsic.

Args
uri The URI of an intrinsic to embed.
arg An optional embedded argument of the call, or None if no argument is supplied.

compute_federated_map

View source

compute_federated_map(
    *fn_args, **fn_kwargs
)

compute_federated_map_all_equal

View source

compute_federated_map_all_equal(
    *fn_args, **fn_kwargs
)

compute_federated_mean

View source

compute_federated_mean(
    *fn_args, **fn_kwargs
)

compute_federated_reduce

View source

compute_federated_reduce(
    *fn_args, **fn_kwargs
)

compute_federated_secure_sum

View source

compute_federated_secure_sum(
    *fn_args, **fn_kwargs
)

compute_federated_sum

View source

compute_federated_sum(
    *fn_args, **fn_kwargs
)

compute_federated_value

View source

compute_federated_value(
    value, type_signature
)

Returns an embedded value for a federated type.

Args
value An object to embed in the executor.
type_signature A tff.Type, the type of value.

compute_federated_value_at_clients

View source

compute_federated_value_at_clients(
    *fn_args, **fn_kwargs
)

compute_federated_value_at_server

View source

compute_federated_value_at_server(
    *fn_args, **fn_kwargs
)

compute_federated_weighted_mean

View source

compute_federated_weighted_mean(
    *fn_args, **fn_kwargs
)

compute_federated_zip_at_clients

View source

compute_federated_zip_at_clients(
    *fn_args, **fn_kwargs
)

compute_federated_zip_at_server

View source

compute_federated_zip_at_server(
    *fn_args, **fn_kwargs
)

factory

View source

@classmethod
factory(
    target_executors: Dict[str, tff.framework.Executor]
)

ingest_value

View source

ingest_value(
    value: Any,
    type_signature: tff.Type
) -> tff.framework.ExecutorValue