XlaSpmdShardToFullShape

XlaSpmdShardToFullShape public final class

Una operación utilizada por el particionador XLA SPMD para cambiar de partición manual a

partición automática. Convierte la entrada en forma de fragmento, particionada manualmente en un tensor de forma completa para ser particionada automáticamente con la misma fragmentación utilizada por la partición manual.

Constantes

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

Métodos públicos

Salida <T>
asOutput ()
Devuelve el identificador simbólico del tensor.
estática <T se extiende Ttype > XlaSpmdShardToFullShape <T>
crear ( Alcance alcance, operando <T> entrada, Cadena manualSharding, Forma fullShape)
Método de fábrica para crear una clase que envuelva una nueva operación XlaSpmdShardToFullShape.
Salida <T>
salida ()

Métodos heredados

Constantes

OP_NAME pública final static String

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "XlaSpmdShardToFullShape"

Métodos públicos

pública de salida <T> asOutput ()

Devuelve el identificador simbólico del 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.

public static XlaSpmdShardToFullShape <T> crear ( Alcance alcance, operando <T> entrada, Cadena manualSharding, Forma fullShape)

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

Parámetros
alcance alcance actual
Devoluciones
  • una nueva instancia de XlaSpmdShardToFullShape

pública de salida <T> salida ()