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> | |
Uscita <T> | produzione () |
Metodi ereditati
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
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