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.