CollateTPUEmbeddingMemory

clase final pública CollateTPUEmbeddingMemory

Una operación que fusiona los protos de configuración de memoria codificados en cadenas de todos los hosts.

Métodos públicos

Salida <Cadena>
como salida ()
Devuelve el identificador simbólico de un tensor.
Intercalación estáticaTPUEmbeddingMemory
crear (alcance alcance , Iterable < Operando <Cadena >> configuraciones de memoria)
Método de fábrica para crear una clase que envuelve una nueva operación CollateTPUEmbeddingMemory.
Salida <Cadena>
fusionadaMemoryConfig ()

Métodos heredados

Métodos públicos

Salida pública <Cadena> 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 CollateTPUEmbeddingMemory (alcance alcance , Iterable< operando <cadena>>memoriaConfigs)

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

Parámetros
alcance alcance actual
configuraciones de memoria Protos de configuración de memoria codificados en cadenas que contienen metadatos sobre las asignaciones de memoria reservadas para TPUEmbedding en todos los hosts.
Devoluciones
  • una nueva instancia de CollateTPUEmbeddingMemory

Salida pública <Cadena> fusionadaMemoryConfig ()