Un'operazione che esegue il partizionatore TPUEmbedding sulla configurazione centrale
dispositivo e calcola la dimensione HBM (in byte) richiesta per l'operazione TPUEmbedding.
Metodi pubblici
Uscita <Stringa> | comeuscita () Restituisce la maniglia simbolica di un tensore. |
Uscita <Stringa> | configurazione comune () Un protocollo di configurazione comune codificato in stringa contenente metadati sull'output del partizionatore TPUEmbedding e la dimensione HBM (in byte) richiesta per il funzionamento. |
statico ExecuteTPUEmbeddingPartitioner |
Metodi ereditati
Metodi pubblici
output pubblico <String> asOutput ()
Restituisce la maniglia simbolica di un 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.
output pubblico <String> commonConfig ()
Un protocollo di configurazione comune codificato in stringa contenente metadati sull'output del partizionatore TPUEmbedding e la dimensione HBM (in byte) richiesta per il funzionamento.
creazione statica pubblica ExecuteTPUEmbeddingPartitioner (ambito ambito , configurazione stringa)
Metodo factory per creare una classe che racchiude una nuova operazione ExecuteTPUEmbeddingPartitioner.
Parametri
scopo | ambito attuale |
---|---|
config | Un protocollo TPUEmbeddingConfiguration serializzato in una stringa, che descrive la configurazione TPUEmbedding desiderata. |
ritorna
- una nuova istanza di ExecuteTPUEmbeddingPartitioner