InfeedEnqueuePrelinearizedBuffer

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
create ( Escopo do escopo, entrada Operando <?>, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação 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.