Clase public final InfeedEnqueuePrelinearizedBuffer
Una operación que pone en cola un búfer prelinealizado en la entrada de TPU.
Clases anidadas
| clase | InfeedEnqueuePrelinearizedBuffer.Options | Los atributos opcionales para InfeedEnqueuePrelinearizedBuffer | |
Métodos públicos
| estática InfeedEnqueuePrelinearizedBuffer | crear ( Alcance alcance, operando <?> entrada, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación InfeedEnqueuePrelinearizedBuffer. |
| estáticas InfeedEnqueuePrelinearizedBuffer.Options | deviceOrdinal (Long deviceOrdinal) |
Métodos heredados
Métodos públicos
public static InfeedEnqueuePrelinearizedBuffer crear ( Alcance alcance, operando <?> entrada, 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. |
|---|