Defined in generated file: python/ops/gen_tpu_ops.py
An op which feeds a single Tensor value into the computation.
tf.contrib.tpu.infeed_enqueue(
input,
shape=[],
layout=[],
device_ordinal=-1,
name=None
)
Args:
input
: ATensor
. A tensor that will be provided using the infeed mechanism.shape
: An optionaltf.TensorShape
or list ofints
. Defaults to[]
. The shape of the tensor.layout
: An optional list ofints
. Defaults to[]
. A vector holding the requested layout in minor-to-major sequence. If a layout attribute is passed, but its values are all -1, the layout will be computed by the infeed operation.device_ordinal
: An optionalint
. 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.