Una operación utilizada por el particionador XLA SPMD para cambiar de la partición automática a
partición manual. Anota la entrada (de forma completa, que se dividirá automáticamente) con la misma fragmentación utilizada en la partición manual y genera un tensor en forma de fragmento que se consumirá en operaciones posteriores con particiones manuales. Si la forma no se puede dividir uniformemente, la región de relleno se enmascarará con ceros.
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 > XlaSpmdFullToShardShape <T> | crear (ámbito de alcance , entrada de operando <T>, fragmentación manual de cadenas) Método de fábrica para crear una clase que envuelve una nueva operación XlaSpmdFullToShardShape. |
Salida <T> | producción () |
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.
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 XlaSpmdFullToShardShape <T> (alcance alcance , entrada operando <T>, cadena manualSharding)
Método de fábrica para crear una clase que envuelve una nueva operación XlaSpmdFullToShardShape.
Parámetros
alcance | alcance actual |
---|
Devoluciones
- una nueva instancia de XlaSpmdFullToShardShape