InfeedEnqueuePrelinearizedBuffer

public clase final InfeedEnqueuePrelinearizedBuffer

Una operación que pone en cola un búfer prelinealizado en la entrada de TPU.

Clases anidadas

clase InfeedEnqueuePrelinearizedBuffer.Options Atributos opcionales para InfeedEnqueuePrelinearizedBuffer

Métodos públicos

static InfeedEnqueuePrelinearizedBuffer
crear ( alcance de alcance, operando <?> entrada, Opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación InfeedEnqueuePrelinearizedBuffer.
static InfeedEnqueuePrelinearizedBuffer.Options
deviceOrdinal (Long deviceOrdinal)

Métodos heredados

Métodos públicos

public static InfeedEnqueuePrelinearizedBuffer create ( alcance del alcance, entrada del operando <?>, Opciones ... opciones)

Método de fábrica para crear una clase que envuelva una nueva operación InfeedEnqueuePrelinearizedBuffer.

Parámetros
alcance alcance actual
aporte Un tensor variante que representa la salida linealizada.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de InfeedEnqueuePrelinearizedBuffer

public static InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Long deviceOrdinal)

Parámetros
deviceOrdinal El dispositivo de TPU que se utilizará. Debe ser -1 cuando el Op se está ejecutando en un dispositivo de TPU y = 0 cuando el Op se está ejecutando en el dispositivo de la CPU.