classe final pública InfeedEnqueuePrelinearizedBuffer
Um op que enfileira o buffer pré-linearizado na alimentação da TPU.
Classes aninhadas
| aula | InfeedEnqueuePrelinearizedBuffer.Options | Atributos opcionais para InfeedEnqueuePrelinearizedBuffer | |
Métodos públicos
| static InfeedEnqueuePrelinearizedBuffer | |
| static InfeedEnqueuePrelinearizedBuffer.Options | deviceOrdinal (Longo deviceOrdinal) |
Métodos Herdados
Métodos públicos
public static InfeedEnqueuePrelinearizedBuffer create ( Escopo do escopo, Operando <?> entrada, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação InfeedEnqueuePrelinearizedBuffer.
Parâmetros
| alcance | escopo atual |
|---|---|
| entrada | Um tensor variante que representa a saída linearizada. |
| opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de InfeedEnqueuePrelinearizedBuffer
public static InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Long deviceOrdinal)
Parâmetros
| dispositivoOrdinal | O dispositivo TPU a ser usado. Deve ser -1 quando o Op estiver sendo executado em um dispositivo TPU e = 0 quando o Op estiver sendo executado no dispositivo CPU. |
|---|