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