tff.framework.ExecutionContext

Stay organized with collections Save and categorize content based on your preferences.

A synchronous execution context backed by an executor_base.Executor.

Inherits From: Context

Used in the notebooks

Used in the tutorials

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.

executor_factory

Methods

invoke

View source

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.