TPUOrdinalSelector

TPUOrdinalSelector classe final pública

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

Output <Integer>
asOutput ()
Retorna o identificador simbólico de um tensor.
estática TPUOrdinalSelector
criar ( Âmbito escopo)
Método de fábrica para criar uma classe que envolve uma nova operação TPUOrdinalSelector.
Output <Integer>
deviceOrdinals ()
Um vetor de 1 ou mais núcleos de TPU.

Métodos herdados

Métodos Públicos

pública 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 criar ( Scope 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

pública Output <Integer> deviceOrdinals ()

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