TPURoundRobin

clase final pública TPURoundRobin

Equilibrio de carga por turnos en núcleos de TPU.

Una operación de equilibrio de carga que realiza operaciones por turnos entre los núcleos de TPU.

Esta operación realiza operaciones por turnos entre los números enteros en [0, NumTPUCoresVisiblePerHost]. Es útil para interactuar con operaciones de TensorFlow que toman como entrada un núcleo de TPU en el que ejecutar cálculos, como `TPUPartitionedCall`.

device_ordinal: un número entero en [0, NumTPUCoresVisiblePerHost].

Métodos públicos

Salida <Entero>
como salida ()
Devuelve el identificador simbólico de un tensor.
TPU estáticoRoundRobin
crear ( alcance alcance)
Método de fábrica para crear una clase que envuelve una nueva operación TPURoundRobin.
Salida <Entero>

Métodos heredados

Métodos públicos

Salida pública <Entero> comoSalida ()

Devuelve el identificador simbólico de un tensor.

Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.

creación pública estática de TPURoundRobin (alcance de alcance )

Método de fábrica para crear una clase que envuelve una nueva operación TPURoundRobin.

Parámetros
alcance alcance actual
Devoluciones
  • una nueva instancia de TPURoundRobin

salida pública <Entero> dispositivoOrdinal ()