Module: tff.tensorflow

Libraries for interacting with a TensorFlow frontend and backend.

Functions

computation(...): Decorates/wraps Python functions and defuns as TFF TensorFlow computations.

replace_intrinsics_with_bodies(...): Iterates over all intrinsic bodies, inlining the intrinsics in comp.

transform_args(...): Transform the arguments to TensorFlow computations.

transform_result(...): Transforms the result of TensorFlow computations.