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],
    local_computation_factory: local_computation_factory_base.LocalComputationFactory = tensorflow_computation_factory.TensorFlowComputationFactory()
)

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.
local_computation_factory An instance of LocalComputationFactory to use to construct local computations used as parameters in certain federated operators (such as tff.federated_sum, etc.). Defaults to a TensorFlow computation factory that generates TensorFlow code.

TypeError If target_executors is not a dict, where each key is a placements.PlacementLiteral and each value is either an executor_base.Executor or a list of executor_base.Executors.
ValueError If target_executors contains a placements.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(
    arg
)

Returns an embedded call for a federated aggregate.

compute_federated_apply

View source

compute_federated_apply(
    arg
)

Returns an embedded call for a federated apply.

compute_federated_broadcast

View source

compute_federated_broadcast(
    arg
)

Returns an embedded call for a federated broadcast.

compute_federated_collect

View source

compute_federated_collect(
    arg
)

Returns an embedded call for a federated collect.

compute_federated_eval_at_clients

View source

compute_federated_eval_at_clients(
    arg
)

Returns an embedded call for a federated eval at clients.

compute_federated_eval_at_server

View source

compute_federated_eval_at_server(
    arg
)

Returns an embedded call for a federated eval at server.

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(
    arg
)

Returns an embedded call for a federated map.

compute_federated_map_all_equal

View source

compute_federated_map_all_equal(
    arg
)

Returns an embedded call for a federated map all equal.

compute_federated_mean

View source

compute_federated_mean(
    arg
)

Returns an embedded call for a federated mean.

compute_federated_secure_select

View source

compute_federated_secure_select(
    arg
)

Returns an embedded call for a federated secure select.

compute_federated_secure_sum

View source

compute_federated_secure_sum(
    arg
)

Returns an embedded call for a federated secure sum.

compute_federated_select

View source

compute_federated_select(
    arg
)

Returns an embedded call for a federated select.

compute_federated_sum

View source

compute_federated_sum(
    arg
)

Returns an embedded call for a federated sum.

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(
    arg
)

Returns an embedded call for a federated value at clients.

compute_federated_value_at_server

View source

compute_federated_value_at_server(
    arg
)

Returns an embedded call for a federated value at server.

compute_federated_weighted_mean

View source

compute_federated_weighted_mean(
    arg
)

Returns an embedded call for a federated weighted mean.

compute_federated_zip_at_clients

View source

compute_federated_zip_at_clients(
    arg
)

Returns an embedded call for a federated zip at clients.

compute_federated_zip_at_server

View source

compute_federated_zip_at_server(
    arg
)

Returns an embedded call for a federated zip at server.

factory

View source

@classmethod
factory(
    target_executors: Dict[str, tff.framework.Executor],
    local_computation_factory: local_computation_factory_base.LocalComputationFactory = tensorflow_computation_factory.TensorFlowComputationFactory()
)

ingest_value

View source

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

reduce

View source

reduce(
    val, zero, op, op_type
)