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