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