tf.raw_ops.InfeedEnqueuePrelinearizedBuffer
An op which enqueues prelinearized buffer into TPU infeed.
tf.raw_ops.InfeedEnqueuePrelinearizedBuffer( input, device_ordinal=-1, name=None )
Args | |
---|---|
input | A Tensor of type variant . A variant tensor representing linearized output. |
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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/InfeedEnqueuePrelinearizedBuffer