TPURoundRobin

classe final pública TPUroundRobin

Balanceamento de carga round-robin em núcleos de TPU.

Uma operação de balanceamento de carga que realiza round-robins entre núcleos de TPU.

Este op round-robins entre os inteiros em [0, NumTPUCoresVisiblePerHost]. É útil para fazer interface com operações do TensorFlow que recebem como entrada um núcleo TPU no qual executar cálculos, como `TPUPartitionedCall`.

device_ordinal: Um inteiro em [0, NumTPUCoresVisiblePerHost].

Métodos públicos

Saída <Inteiro>
comoSaída ()
Retorna o identificador simbólico de um tensor.
estático TPUroundRobin
criar ( escopo do escopo)
Método de fábrica para criar uma classe que envolve uma nova operação TPUroundRobin.
Saída <Inteiro>

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 TPUroundRobin create ( escopo do escopo)

Método de fábrica para criar uma classe que envolve uma nova operação TPUroundRobin.

Parâmetros
alcance escopo atual
Devoluções
  • uma nova instância de TPUroundRobin

Public Output <Integer> deviceOrdinal ()