InfeedEnqueuePrelinearizedBuffer

clase final pública InfeedEnqueuePrelinearizedBuffer

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

Clases anidadas

clase InfeedEnqueuePrelinearizedBuffer.Options Atributos opcionales para InfeedEnqueuePrelinearizedBuffer

Métodos públicos

InfeedEnqueue estáticoBuffer prelinealizado
crear (alcance alcance , entrada de operando <?>, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación InfeedEnqueuePrelinearizedBuffer.
Estático InfeedEnqueuePrelinearizedBuffer.Options
dispositivoOrdinal (dispositivo largoOrdinal)

Métodos heredados

Métodos públicos

creación pública estática de InfeedEnqueuePrelinearizedBuffer (alcance del alcance , entrada del operando <?>, opciones... opciones)

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

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

público estático InfeedEnqueuePrelinearizedBuffer.Options dispositivoOrdinal (dispositivo largoOrdinal)

Parámetros
dispositivoOrdinal El dispositivo TPU a utilizar. Esto debe ser -1 cuando el Op se ejecuta en un dispositivo TPU y = 0 cuando el Op se ejecuta en el dispositivo CPU.