tff.backends.native.transform_to_native_form

Compiles a computation for execution in the TFF native runtime.

tff.backends.native.transform_to_native_form(
    comp: tff.framework.ConcreteComputation,
    transform_math_to_tf: bool = False,
    grappler_config: Optional[tf.compat.v1.ConfigProto] = None
) -> tff.framework.ConcreteComputation

This function transforms the proto underlying comp by transforming it to call-dominant form (see tff.framework.to_call_dominant for definition).

comp Instance of computation_impl.ConcreteComputation to compile.
transform_math_to_tf Whether to additional transform math to TensorFlow graphs. Necessary if running on a execution state without ReferenceResolvingExecutors underneath FederatingExecutors.
grappler_config Configuration for Grappler optimizations to perform on the TensorFlow computations. If None, Grappler will not be run and no optimizations wil be applied.

A new computation_impl.ConcreteComputation representing the compiled version of comp.