InfeedEnqueuePrelinearizedBuffer

public final class InfeedEnqueuePrelinearizedBuffer

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

Classes aninhadas

classe InfeedEnqueuePrelinearizedBuffer.Options Atributos opcionais para InfeedEnqueuePrelinearizedBuffer

Métodos Públicos

estática 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.
estáticos 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.