OrdinalSelector

clase final pública OrdinalSelector

Un selector de núcleo de TPU Op.

Esta operación produce un conjunto de núcleos de TPU (para calentamiento) o un único núcleo de TPU (para inferencia regular) para ejecutar el programa de TPU. La salida la consume TPUPartitionedCall.

Constantes

Cadena OP_NOMBRE El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Métodos públicos

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

Métodos heredados

Constantes

Cadena final estática pública OP_NAME

El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Valor constante: "TPUOrdinalSelector"

Métodos públicos

Salida pública < TInt32 > asOutput ()

Devuelve el identificador simbólico del 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 OrdinalSelector ( alcance alcance)

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

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

Salida pública < TInt32 > deviceOrdinals ()

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