InfeedEnqueuePrelinearizedBuffer

clase final pública InfeedEnqueuePrelinearizedBuffer

Una operación que pone en cola el 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 alcance, Operando <?> entrada, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación InfeedEnqueuePrelinearizedBuffer.
static InfeedEnqueuePrelinearizedBuffer.Options
deviceOrdinal (dispositivo largo ordinal)

Métodos Heredados

Métodos públicos

public static InfeedEnqueuePrelinearizedBuffer create ( Ámbito de alcance, Operando <?> entrada, Opciones... opciones)

Método de fábrica para crear una clase que envuelve 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 dispositivo ordinal ( dispositivo largo ordinal)

Parámetros
dispositivoordinal El dispositivo de TPU a utilizar. Debe ser -1 cuando la operación se ejecuta en un dispositivo de TPU y = 0 cuando la operación se ejecuta en el dispositivo de CPU.