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
Saída <Integer> | asOutput () Retorna o identificador simbólico de um tensor. |
TPUOrdinalSelector estático | |
Saída <Integer> | deviceOrdinals () Um vetor de 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 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 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