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