Uma operação usada pelo particionador XLA SPMD para mudar do particionamento automático para
particionamento manual. Ele anota a entrada (formato completo, para ser particionado automaticamente) com o mesmo sharding usado pelo particionamento manual e gera um tensor em forma de shard para ser consumido por operações particionadas manualmente posteriormente. Se a forma não for particionável uniformemente, a região de preenchimento será mascarada com 0s.
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
Saída <T> | asOutput () Retorna o identificador simbólico do tensor. |
estático <T estende TType > XlaSpmdFullToShardShape <T> | create (escopo do escopo , entrada do operando <T>, string manualSharding) Método de fábrica para criar uma classe que agrupa uma nova operação XlaSpmdFullToShardShape. |
Saída <T> | saída () |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
Saída pública <T> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para 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 (escopo do escopo , entrada Operand <T>, String manualSharding)
Método de fábrica para criar uma classe que agrupa uma nova operação XlaSpmdFullToShardShape.
Parâmetros
escopo | escopo atual |
---|
Devoluções
- uma nova instância de XlaSpmdFullToShardShape