XlaSpmdShardToFullShape

classe final pública XlaSpmdShardToFullShape

Uma operação usada pelo particionador XLA SPMD para alternar do particionamento manual para

particionamento automático. Ele converte a entrada particionada manualmente em formato de fragmento em um tensor de formato completo para ser particionado automaticamente com o mesmo fragmento usado pelo particionamento manual.

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 > XlaSpmdShardToFullShape <T>
create ( Escopo , entrada Operando <T>, String manualSharding, Shape fullShape)
Método de fábrica para criar uma classe que envolve uma nova operação XlaSpmdShardToFullShape.
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

Valor Constante: "XlaSpmdShardToFullShape"

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 XlaSpmdShardToFullShape <T> create (escopo do escopo , entrada do operando <T>, String manualSharding, Shape fullShape)

Método de fábrica para criar uma classe que envolve uma nova operação XlaSpmdShardToFullShape.

Parâmetros
escopo escopo atual
Devoluções
  • uma nova instância de XlaSpmdShardToFullShape

Saída pública <T> saída ()