OrdinalSelector

OrdinalSelector public final class

Un selector de núcleo de TPU Op.

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

Constantes

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

Métodos públicos

Salida < TInt32 >
asOutput ()
Devuelve el identificador simbólico del tensor.
estática OrdinalSelector
crear ( Alcance alcance)
Método de fábrica para crear una clase que envuelva una nueva operación OrdinalSelector.
Salida < TInt32 >
deviceOrdinals ()
Un vector de 1 o más núcleos de TPU.

Métodos heredados

Constantes

OP_NAME pública final static String

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

Valor constante: "TPUOrdinalSelector"

Métodos públicos

pública de salida < 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.

public static OrdinalSelector crear ( Alcance alcance)

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

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

pública de salida < TInt32 > deviceOrdinals ()

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