tff.backends.native.transform_to_native_form

See Stable See Nightly

Compiles a computation for execution in the TFF native runtime.

tff.backends.native.transform_to_native_form(
    comp: tff.Computation
) -> tff.Computation

This function transforms the proto underlying comp by first transforming it to call-dominant form (see tff.framework.transform_to_call_dominant for definition), then computing information on the dependency structure of the bindings and remapping them into tuples, such that every computation is evaluated as early as possible, and parallelized with any other computation with which it shares dependency structure.

Args
comp	Instance of computation_base.Computation to compile.

Returns
A new computation_base.Computation representing the compiled version of comp.