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 | |
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