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.
Métodos públicos
Salida <Entero> | como salida () Devuelve el identificador simbólico de un tensor. |
TPUOrdinalSelector estático | |
Salida <Entero> | dispositivoOrdinales () Un vector 1 o más núcleos de TPU. |
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 TPUOrdinalSelector (alcance de 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