View source on GitHub
|
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.
View source on GitHub