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.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2021-08-16 UTC.
[null,null,["Last updated 2021-08-16 UTC."],[],[]]