TPUOrdinalSelector

TPUOrdinalSelector public final class

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 inferencias regulares) para ejecutar el programa de TPU. TPUPartitionedCall consume la salida.

Métodos públicos

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

Métodos heredados

Métodos públicos

pública de salida <Integer> asOutput ()

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.

public static TPUOrdinalSelector crear ( Alcance alcance)

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

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

pública de salida <Integer> deviceOrdinals ()

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