Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.tpu.outside_compilation

View source on GitHub

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

Aliases:

  • tf.compat.v1.tpu.outside_compilation
  • tf.contrib.tpu.outside_compilation
tf.tpu.outside_compilation(
    computation,
    *args,
    **kwargs
)

Args:

  • 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.

Returns:

The Tensors returned by computation.