TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.tpu.outfeed_dequeue

Defined in generated file: python/ops/gen_tpu_ops.py

Retrieves a single tensor from the computation outfeed.

tf.contrib.tpu.outfeed_dequeue(
    dtype,
    shape,
    device_ordinal=-1,
    name=None
)

This operation will block indefinitely until data is available.

Args:

  • dtype: A tf.DType. The type of elements in the tensor.
  • shape: A tf.TensorShape or list of ints. The shape of the tensor.
  • device_ordinal: An optional int. Defaults to -1. The TPU device to use. This should be -1 when the Op is running on a TPU device, and >= 0 when the Op is running on the CPU device.
  • name: A name for the operation (optional).

Returns:

A Tensor of type dtype.