tf.contrib.tpu.infeed_enqueue

tf.contrib.tpu.infeed_enqueue(
    input,
    shape=[],
    device_ordinal=-1,
    name=None
)

Defined in generated file: tensorflow/contrib/tpu/ops/gen_tpu_ops.py.

An op which feeds a single Tensor value into the computation.

Args:

  • input: A Tensor. A tensor that will be provided using the infeed mechanism.
  • shape: An optional tf.TensorShape or list of ints. Defaults to []. 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:

The created Operation.