TPURoundRobin

classe final pública TPURoundRobin

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
criar (escopo do escopo )
Método de fábrica para criar uma classe que envolve uma nova operação TPURoundRobin.
Saída <Inteiro>
dispositivoOrdinal ()

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

Saída pública <Inteiro> deviceOrdinal ()