Balanceamento de carga round-robin em núcleos de TPU.
Uma operação de balanceamento de carga que faz round-robins entre núcleos de TPU.
Esta operação faz round-robins entre os inteiros em [0, NumTPUCoresVisiblePerHost]. É útil para fazer interface com operações do TensorFlow que usam como entrada um núcleo TPU no qual executar cálculos, como `TPUPartitionedCall`.
device_ordinal: um número inteiro em [0, NumTPUCoresVisiblePerHost].
Métodos Públicos
Saída <Inteiro> | asOutput () Retorna o identificador simbólico de um tensor. |
TPURoundRobin estático | |
Saída <Inteiro> |
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.
criação pública estática de TPURoundRobin (escopo do escopo )
Método de fábrica para criar uma classe que envolve uma nova operação TPURoundRobin.
Parâmetros
escopo | escopo atual |
---|
Devoluções
- uma nova instância de TPURoundRobin