XlaSpmdFullToShardShape

XlaSpmdFullToShardShape public final class

Una operación utilizada por el particionador XLA SPMD para cambiar del particionamiento automático a

particionamiento manual. Anota la entrada (forma completa, para ser particionada automáticamente) con la misma fragmentación utilizada por la partición manual, y genera un tensor en forma de fragmento para ser consumido por operaciones posteriores particionadas manualmente. Si la forma no se puede dividir uniformemente, la región de relleno se enmascarará con ceros.

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 > XlaSpmdFullToShardShape <T>
crear ( Alcance alcance, operando <T> entrada, Cadena manualSharding)
Método de fábrica para crear una clase que envuelva una nueva operación XlaSpmdFullToShardShape.
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: "XlaSpmdFullToShardShape"

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 XlaSpmdFullToShardShape <T> crear ( Alcance alcance, operando <T> entrada, Cadena manualSharding)

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

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

pública de salida <T> salida ()