XlaSpmdFullToShardShape

classe finale pubblica XlaSpmdFullToShardShape

Un'operazione utilizzata dal partizionatore XLA SPMD per passare dal partizionamento automatico a

partizionamento manuale. Annota l'input (a forma completa, da partizionare automaticamente) con lo stesso sharding utilizzato dal partizionamento manuale e restituisce un tensore a forma di shard da utilizzare per le operazioni successive partizionate manualmente. Se la forma non è partizionabile in modo uniforme, la regione di riempimento verrà mascherata con 0.

Costanti

Corda OP_NAME Il nome di questa operazione, come noto al motore principale di TensorFlow

Metodi pubblici

Uscita <T>
comeuscita ()
Restituisce l'handle simbolico del tensore.
statico <T estende TType > XlaSpmdFullToShardShape <T>
create (ambito ambito , input operando <T>, stringa manualeSharding)
Metodo factory per creare una classe che racchiude una nuova operazione XlaSpmdFullToShardShape.
Uscita <T>
produzione ()

Metodi ereditati

Costanti

Stringa finale statica pubblica OP_NAME

Il nome di questa operazione, come noto al motore principale di TensorFlow

Valore costante: "XlaSpmdFullToShardShape"

Metodi pubblici

Uscita pubblica <T> asOutput ()

Restituisce l'handle simbolico del tensore.

Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.

public static XlaSpmdFullToShardShape <T> create ( scope scope, operando <T> input, string manualSharding)

Metodo factory per creare una classe che racchiude una nuova operazione XlaSpmdFullToShardShape.

Parametri
scopo ambito attuale
ritorna
  • una nuova istanza di XlaSpmdFullToShardShape

Uscita pubblica <T> uscita ()