View source on GitHub
|
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_local_async_python_execution_context(...): Creates a context that executes computations locally as coro functions.
create_local_python_execution_context(...): Creates an execution context that executes computations locally.
create_mergeable_comp_execution_context(...): Creates context which compiles to and executes mergeable comp form.
create_remote_python_execution_context(...): Creates context to execute computations with workers on channels.
create_sizing_execution_context(...): Creates an execution context that executes computations locally.
create_sync_local_cpp_execution_context(...): Returns an execution context backed by C++ runtime.
create_thread_debugging_execution_context(...): Creates a simple execution context that executes computations locally.
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_local_async_python_execution_context(...): Sets a context that executes computations locally as coro functions.
set_local_python_execution_context(...): Sets an execution context that executes computations locally.
set_mergeable_comp_execution_context(...): Sets context which compiles to and executes mergeable comp form.
set_remote_python_execution_context(...): Installs context to execute computations with workers on channels.
set_sync_local_cpp_execution_context(...): Sets default context to a C++ runtime.
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.