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. |
---|