TPUOrdinalSelector

public final class TPUOrdinalSelector

Um seletor de núcleo TPU Op.

Este Op produz um conjunto de núcleos de TPU (para aquecimento) ou um único núcleo de TPU (para inferência regular) para executar o programa de TPU. A saída é consumida por TPUPartitionedCall.

Métodos Públicos

Saída <Integer>
asOutput ()
Retorna o identificador simbólico de um tensor.
TPUOrdinalSelector estático
criar ( escopo do escopo)
Método de fábrica para criar uma classe envolvendo uma nova operação TPUOrdinalSelector.
Saída <Integer>
deviceOrdinals ()
Um vetor de 1 ou mais núcleos de TPU.

Métodos herdados

Métodos Públicos

Public Output <Integer> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.

public static TPUOrdinalSelector create ( escopo do escopo)

Método de fábrica para criar uma classe envolvendo uma nova operação TPUOrdinalSelector.

Parâmetros
alcance escopo atual
Devoluções
  • uma nova instância de TPUOrdinalSelector

public Output <Integer> deviceOrdinals ()

Um vetor de 1 ou mais núcleos de TPU.