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 |
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