ExecuteTPUEmbeddingPartitioner

clase final pública ExecuteTPUEmbeddingPartitioner

Una operación que ejecuta el particionador TPUEmbedding en la configuración central

dispositivo y calcula el tamaño de HBM (en bytes) necesario para la operación TPUEmbedding.

Métodos públicos

Salida <Cadena>
como salida ()
Devuelve el identificador simbólico de un tensor.
Salida <Cadena>
configuración común ()
Un protocolo de configuración común codificado en cadena que contiene metadatos sobre la salida del particionador TPUEmbedding y el tamaño de HBM (en bytes) necesario para la operación.
Ejecutar estáticoTPUEmbeddingPartitioner
crear (alcance alcance , configuración de cadena)
Método de fábrica para crear una clase que envuelve una nueva operación ExecuteTPUEmbeddingPartitioner.

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.

Salida pública <Cadena> commonConfig ()

Un protocolo de configuración común codificado en cadena que contiene metadatos sobre la salida del particionador TPUEmbedding y el tamaño de HBM (en bytes) necesario para la operación.

creación estática pública de ExecuteTPUEmbeddingPartitioner (alcance del alcance , configuración de cadena)

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

Parámetros
alcance alcance actual
configuración Un prototipo de TPUEmbeddingConfiguration serializado en una cadena, que describe la configuración de TPUEmbedding deseada.
Devoluciones
  • una nueva instancia de ExecuteTPUEmbeddingPartitioner