Una operación que ejecuta el particionador TPUEmbedding en la configuración central
dispositivo y calcula el tamaño 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 prototipo de configuración común codificado en cadena que contiene metadatos sobre la salida del particionador TPUEmbedding y el tamaño HBM (en bytes) requerido para la operación. |
estático ExecuteTPUEmbeddingPartitioner |
Métodos Heredados
Métodos públicos
Salida pública <String> 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.
Salida pública <String> commonConfig ()
Un prototipo de configuración común codificado en cadena que contiene metadatos sobre la salida del particionador TPUEmbedding y el tamaño HBM (en bytes) requerido para la operación.
public static ExecuteTPUEmbeddingPartitioner create ( Scope scope, String config)
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