TPUOrdinalSelector

classe final pública TPUOrdinalSelector

Um seletor de núcleo TPU Op.

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

Métodos públicos

Saída <Inteiro>
comoSaída ()
Retorna o identificador simbólico de um tensor.
estático TPUOrdinalSelector
criar ( escopo do escopo)
Método de fábrica para criar uma classe que envolve uma nova operação TPUOrdinalSelector.
Saída <Inteiro>
deviceOrdinais ()
Um vetor 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 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 que envolve uma nova operação TPUOrdinalSelector.

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

Public Output <Integer> deviceOrdinals ()

Um vetor 1 ou mais núcleos de TPU.