tff.program.NativeFederatedContext

A tff.program.FederatedContext backed by a tff.framework.Context.

Inherits From: FederatedContext, Context

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

context A tff.framework.Context with an async invoke.

ValueError If context does not have an async invoke.

Methods

invoke

View source

invoke(
    comp: tff.Computation,
    arg: Any
) -> Any

Invokes the comp with the argument arg.

Args
comp The tff.Computation being invoked.
arg The optional argument of comp.

Returns
The result of invocation, must contain only structures, server-placed values, or tensors.

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