InfeedEnqueuePrelinearizedBuffer

classe final pública InfeedEnqueuePrelinearizedBuffer

Um op que enfileira o buffer pré-linearizado na alimentação de TPU.

Classes aninhadas

aula InfeedEnqueuePrelinearizedBuffer.Options Atributos opcionais para InfeedEnqueuePrelinearizedBuffer

Métodos Públicos

static InfeedEnqueuePrelinearizedBuffer
criar ( escopo do escopo, entrada Operand <?>, Opções ... opções)
Método de fábrica para criar uma classe que envolve uma nova operação InfeedEnqueuePrelinearizedBuffer.
static InfeedEnqueuePrelinearizedBuffer.Options
deviceOrdinal (Long deviceOrdinal)

Métodos herdados

Métodos Públicos

public static InfeedEnqueuePrelinearizedBuffer criar ( Scope escopo, Operando <?> entrada, Options ... 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
deviceOrdinal O dispositivo TPU a ser usado. Deve ser -1 quando o Op está em execução em um dispositivo TPU e = 0 quando o Op está em execução no dispositivo de CPU.