ConfigureTPUEmbeddingMemory

classe finale pubblica ConfigureTPUEmbeddingMemory

Un'operazione che configura il software TPUEmbedding su un host.

Metodi pubblici

Uscita <Stringa>
comeuscita ()
Restituisce la maniglia simbolica di un tensore.
static ConfigureTPUEmbeddingMemory
create ( ambito ambito , operando <String> commonConfig)
Metodo factory per creare una classe che racchiude una nuova operazione ConfigureTPUEmbeddingMemory.
Uscita <Stringa>
memoriaConfigurazione ()
Una configurazione di memoria con codifica stringa contenente metadati sulle allocazioni di memoria riservate per TPUEmbedding.

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.

public static ConfigureTPUEmbeddingMemory create ( ambito ambito , operando <String> commonConfig)

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

Parametri
scopo ambito attuale
commonConfig Un protocollo CommonConfiguration codificato in stringa contenente metadati sull'output del partizionatore TPUEmbedding e la dimensione HBM (in byte) richiesta per il funzionamento.
ritorna
  • una nuova istanza di ConfigureTPUEmbeddingMemory

Uscita pubblica <String> memoryConfig ()

Una configurazione di memoria con codifica stringa contenente metadati sulle allocazioni di memoria riservate per TPUEmbedding.