Google I/O returns May 18-20! Reserve space and build your schedule Register now

Module: tff.experimental.backends.xla

Libraries for interacting with XLA backends.

Functions

create_local_execution_context(...): Creates an XLA-based local execution context.

set_local_execution_context(...): Sets an XLA-based local execution context.