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

View source on GitHub

A placeholder op for a value that will be fed into the computation.

tf.contrib.tpu.infeed_dequeue(
    dtype,
    shape,
    name=None
)

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.
  • name: A name for the operation (optional).

Returns:

A Tensor of type dtype. A tensor that will be provided using the infeed mechanism.

Raises:

  • TypeError: If 'dtype` is not a supported infeed type.