tff.experimental.backends.xla.set_local_execution_context

Sets an XLA-based local execution context.

tff.experimental.backends.xla.set_local_execution_context(
    *args, **kwargs
)

Invokes create_local_execution_context to contruct an execution context, and sets it as the default. Accepts the same parameters as create_local_execution_context.

Args
*args	Positional args for create_local_execution_context.
**kwargs	Keyword args for create_local_execution_context.