Watch talks from the 2019 TensorFlow Dev Summit Watch now

tf.contrib.tpu.outside_compilation

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

Defined in tensorflow/contrib/tpu/python/tpu/tpu.py.

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

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.