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 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
crear ( alcance de 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 <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