InfeedEnqueuePrelinearizedBuffer

classe final pública InfeedEnqueuePrelinearizedBuffer

Uma operação que enfileira o buffer pré-linearizado na alimentação da TPU.

Classes aninhadas

aula InfeedEnqueuePrelinearizedBuffer.Options Atributos opcionais para InfeedEnqueuePrelinearizedBuffer

Métodos Públicos

InfeedEnqueuePrelinearizedBuffer estático
create ( Escopo , Operando <?> entrada, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação InfeedEnqueuePrelinearizedBuffer.
InfeedEnqueuePrelinearizedBuffer.Options estático
deviceOrdinal (dispositivo longo Ordinal)

Métodos herdados

Métodos Públicos

public static InfeedEnqueuePrelinearizedBuffer create (escopo do escopo , entrada do operando <?>, opções... opções)

Método de fábrica para criar uma classe que agrupa uma nova operação InfeedEnqueuePrelinearizedBuffer.

Parâmetros
escopo 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.