tf.tpu.outside_compilation

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.