Module: tff.backends.native

Libraries for interacting with native backends.

Functions

compile_to_mergeable_comp_form(...): Compiles a computation with a single aggregation to MergeableCompForm.

create_async_local_cpp_execution_context(...): Returns an execution context backed by C++ runtime.

create_mergeable_comp_execution_context(...): Creates context which compiles to and executes mergeable comp form.

create_sync_local_cpp_execution_context(...): Returns an execution context backed by C++ runtime.

desugar_and_transform_to_native(...): Transform to native form and replace intrinsics with TensorFlow.

set_async_local_cpp_execution_context(...): Sets default context to a C++ runtime.

set_mergeable_comp_execution_context(...): Sets context which compiles to and executes mergeable comp form.

set_sync_local_cpp_execution_context(...): Sets default context to a C++ runtime.

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