tff.backends.test.set_async_experimental_distributed_cpp_execution_context

Sets a local execution context backed by TFF-C++ runtime.

tff.backends.test.set_async_experimental_distributed_cpp_execution_context(
    distributed_config: tff.backends.test.DistributedConfiguration,
    default_num_clients: int = 0,
    max_concurrent_computation_calls: int = -1,
    stream_structs: bool = False
) -> None

distributed_config A runtime configuration for running TF computation in a distributed manner. A server side and/or client side mesh can be supplied in the configuration if TF computation should be executed with DTensor executor.
default_num_clients The number of clients to use as the default cardinality, if thus number cannot be inferred by the arguments of a computation.
max_concurrent_computation_calls The maximum number of concurrent calls to a single computation in the CPP runtime. If nonpositive, there is no limit.
stream_structs The flag to enable decomposing and streaming struct values.