tff.framework.AsyncExecutionContext

View source on GitHub

An asynchronous execution context backed by an executor_base.Executor.

Inherits From: AsyncContext

tff.framework.AsyncExecutionContext(
    executor_fn: tff.framework.ExecutorFactory,
    compiler_fn: Optional[Callable[[_Computation], object]] = None,
    *,
    cardinality_inference_fn: cardinalities_utils.CardinalityInferenceFnType = cardinalities_utils.infer_cardinalities
)

This context's ingest and invoke methods return Python coroutine objects which represent the actual work of ingestion and invocation in the backing executor.

This context will support concurrent invocation of multiple computations if their arguments have the same cardinalities.

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. tff.framework.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 tff.structure.Structs.

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