TPUOrdinalSelector

clase final pública TPUOrdinalSelector

Un selector de núcleo de TPU Op.

Esta operación produce un conjunto de núcleos de TPU (para calentamiento) o un solo núcleo de TPU (para inferencia regular) para ejecutar el programa de TPU. La salida es consumida por TPUPartitionedCall.

Métodos públicos

Salida <Entero>
como salida ()
Devuelve el identificador simbólico de un tensor.
TPUOrdinalSelector estático
crear ( Alcance alcance)
Método de fábrica para crear una clase que envuelve una nueva operación TPUOrdinalSelector.
Salida <Entero>
dispositivoOrdinales ()
Un vector 1 o más núcleos de TPU.

Métodos Heredados

Métodos públicos

salida pública <entero> asOutput ()

Devuelve el identificador simbólico de un tensor.

Las entradas de 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.

TPUOrdinalSelector estático público crear ( alcance del alcance)

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

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

Salida pública <Integer> deviceOrdinals ()

Un vector 1 o más núcleos de TPU.