Compiles all mathematical functions in comp to TensorFlow blocks.
tff.backends.native.transform_mathematical_functions_to_tensorflow(
comp: tff.Computation
) -> tff.Computation
Notice that this does not necessarily represent a strict performance
improvement. In particular, this compilation will not attempt to deduplicate
across the boundaries of communication operators, and therefore it may be
the case that compiling eagerly to TensorFlow hides the opportunity for
a dynamic cache to be used.
Args |
|---|
comp
|
Instance of computation_base.Computation to compile.
|
Returns |
|---|
A new computation_base.Computation representing the compiled version of
comp.
|
View source on GitHub