tff.framework.ExecutionContext

A synchronous execution context backed by an executor_base.Executor.

Inherits From: Context

tff.framework.ExecutionContext(
    executor_fn: tff.framework.ExecutorFactory,
    compiler_fn: Optional[Callable[[computation_base.Computation], Any]] = None,
    *,
    cardinality_inference_fn: cardinalities_utils.CardinalityInferenceFnType = cardinalities_utils.infer_cardinalities
)

Used in the notebooks

Used in the tutorials
Custom Federated Algorithms, Part 2: Implementing Federated Averaging Loading Remote Data in TFF TFF for Federated Learning Research: Model and Update Compression

Args
executor_fn	Instance of executor_factory.ExecutorFactory.
compiler_fn	A Python function that will be used to compile a computation.
cardinality_inference_fn	A Python function specifying how to infer cardinalities from arguments (and their associated types). The value returned by this function will be passed to the create_executor method of executor_fn to construct a tff.framework.Executor instance.

Attributes
executor_factory

Methods

invoke

View source

invoke(
    comp, arg
)

Invokes computation comp with argument arg.

Args
comp	The computation being invoked. The Python type of comp expected here (e.g., pb.Computation. ConcreteComputation, or other) may depend on the context. It is the responsibility of the concrete implementation of this interface to verify that the type of comp matches what the context is expecting.
arg	The argument passed to the computation. If no argument is passed, this will be None. Structural argument types will be normalized into structure.Structs.

Returns
The result of invocation, which is context-dependent.