tff.program.NativeFederatedContext

A tff.program.FederatedContext backed by an execution context.

Inherits From: FederatedContext, SyncContext

tff.program.NativeFederatedContext(
    context: tff.framework.AsyncExecutionContext
)

context An tff.framework.AsyncExecutionContext.

Methods

invoke

View source

invoke(
    comp: tff.Computation,
    arg: Optional[Union[value_reference.MaterializableStructure, Any,
        computation_base.Computation]]
) -> structure_utils.Structure[AwaitableValueReference]

Invokes the comp with the argument arg.

Args
comp The tff.Computation being invoked.
arg The optional argument of comp; server-placed values must be represented by tff.program.MaterializableStructure, and client-placed values must be represented by structures of values returned by a tff.program.FederatedDataSourceIterator.

Returns
The result of invocation; a structure of tff.program.MaterializableValueReference.

Raises
ValueError If the result type of the invoked computation does not contain only structures, server-placed values, or tensors.

Raises
ValueError If the result type of comp does not contain only structures, server-placed values, or tensors.