ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


View source on GitHub

Builds part of a computation outside any current TPU replicate scope.

computation A Python function that builds the computation to place on the host.
*args the positional arguments for the computation.
**kwargs the keyword arguments for the computation.

The Tensors returned by computation.