tff.backends.xla.set_local_python_execution_context

Sets an XLA-based local execution context.

tff.backends.xla.set_local_python_execution_context(
    *args, **kwargs
)

Used in the notebooks

Used in the tutorials
Experimental support for JAX in TFF

Invokes create_local_execution_context to construct 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.