Um op usado pelo particionador XLA SPMD para mudar de particionamento automático para
particionamento manual. Ele anota a entrada (forma completa, para ser particionado automaticamente) com o mesmo fragmento usado pelo particionamento manual e gera um tensor em forma de fragmento para ser consumido por operações particionadas manualmente posteriores. Se a forma não puder ser particionada de maneira uniforme, a região de preenchimento será mascarada com 0s.
Constantes
Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
Output <T> | asOutput () Retorna o identificador simbólico do tensor. |
estática <T estende TType > XlaSpmdFullToShardShape <T> | |
Output <T> | saída () |
Métodos herdados
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Métodos Públicos
pública Output <T> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static XlaSpmdFullToShardShape <T> create ( Scope escopo, Operando <T> de entrada, String manualSharding)
Método de fábrica para criar uma classe envolvendo uma nova operação XlaSpmdFullToShardShape.
Parâmetros
alcance | escopo atual |
---|
Devoluções
- uma nova instância de XlaSpmdFullToShardShape