XlaRecvTPUEmbeddingDeduplicationData

clase final pública XlaRecvTPUEmbeddingDeduplicationData

Recibe datos de deduplicación (índices y pesos) del núcleo de incrustación.

Los datos de deduplicación son un tensor con type=DT_VARIANT. El tensor en sí es una tupla anidada XLA que contiene N elementos (donde N es la relación entre el número de incrustaciones y núcleos de tensor por chip de TPU). Cada elemento de la tupla anidada es una tupla de tensores de rango 1. Cada tensor contiene índices (DT_UINT32) para incrustar la búsqueda en TensorCore o pesos (DT_FLOAT) para aplicar a la salida de la operación de incrustación de búsqueda.

Métodos públicos

Salida <Objeto>
como salida ()
Devuelve el identificador simbólico de un tensor.
estático XlaRecvTPUEmbeddingDeduplicationData
crear ( alcance de alcance, configuración de cadena)
Método de fábrica para crear una clase que envuelve una nueva operación XlaRecvTPUEmbeddingDeduplicationData.
Salida <?>
salida ()

Métodos Heredados

Métodos públicos

Salida pública <Objeto> 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.

Public static XlaRecvTPUEmbeddingDeduplicationData create ( Scope scope, String config)

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

Parámetros
alcance alcance actual
configuración Proto de configuración de integración de TPU serializado.
Devoluciones
  • una nueva instancia de XlaRecvTPUEmbeddingDeduplicationData

salida pública <?> salida ()