TPUOrdinalSelector

classe final pública TPUOrdinalSelector

Um seletor de núcleo TPU Op.

Esta operação 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>
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 que envolve uma nova operação TPUOrdinalSelector.
Saída <Inteiro>
dispositivoOrdinais ()
Um vetor de 1 ou mais núcleos de TPU.

Métodos herdados

Métodos Públicos

Saída pública <Inteiro> 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
escopo escopo atual
Devoluções
  • uma nova instância de TPUOrdinalSelector

Saída pública <Inteiro> deviceOrdinals ()

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