Module: tff.backends.native

See Stable See Nightly

Libraries for interacting with native backends.

Functions

create_local_execution_context(...): Creates an execution context that executes computations locally.

create_remote_execution_context(...): Creates context to execute computations with workers on channels.

create_sizing_execution_context(...): Creates an execution context that executes computations locally.

create_thread_debugging_execution_context(...): Creates a simple execution context that executes computations locally.

set_local_execution_context(...): Sets an execution context that executes computations locally.

set_remote_execution_context(...): Installs context to execute computations with workers on channels.

set_thread_debugging_execution_context(...): Sets an execution context that executes computations locally.

transform_to_native_form(...): Compiles a computation for execution in the TFF native runtime.