Google I/O returns May 18-20! Reserve space and build your schedule Register now


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.