ExecuteTPUEmbeddingPartitioner

classe finale pubblica ExecuteTPUEmbeddingPartitioner

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
creare (ambito ambito , configurazione stringa)
Metodo factory per creare una classe che racchiude una nuova operazione 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