XlaSpmdShardToFullShape

clase final pública XlaSpmdShardToFullShape

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

partición automática. Convierte la entrada dividida manualmente en forma de fragmento en un tensor de forma completa que se dividirá automáticamente con la misma fragmentación utilizada en la partición manual.

Constantes

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

Métodos públicos

Salida <T>
como salida ()
Devuelve el identificador simbólico del tensor.
estático <T extiende TType > XlaSpmdShardToFullShape <T>
crear (alcance del alcance , entrada del operando <T>, fragmentación manual de cadenas, forma completa)
Método de fábrica para crear una clase que envuelve una nueva operación XlaSpmdShardToFullShape.
Salida <T>

Métodos heredados

Constantes

Cadena final estática pública OP_NAME

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

Valor constante: "XlaSpmdShardToFullShape"

Métodos públicos

Salida pública <T> como Salida ()

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.

creación pública estática XlaSpmdShardToFullShape <T> (alcance alcance , entrada operando <T>, cadena manualSharding, forma forma completa)

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

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

Salida pública <T> salida ()