Uma operação que executa o particionador TPUEmbedding na configuração central
dispositivo e calcula o tamanho do HBM (em bytes) necessário para a operação do TPUEmbedding.
Métodos Públicos
Saída <String> | asOutput () Retorna o identificador simbólico de um tensor. |
Saída <String> | commonConfig () Um protocolo de configuração comum codificado em string contendo metadados sobre a saída do particionador TPUEmbedding e o tamanho do HBM (em bytes) necessário para operação. |
ExecuteTPUEmbeddingPartitioner estático |
Métodos herdados
Métodos Públicos
Saída pública <String> asOutput ()
Retorna o identificador simbólico de um tensor.
As entradas para operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
Saída pública <String> commonConfig ()
Um protocolo de configuração comum codificado em string contendo metadados sobre a saída do particionador TPUEmbedding e o tamanho do HBM (em bytes) necessário para operação.
public static ExecuteTPUEmbeddingPartitioner create (escopo do escopo , configuração de string)
Método de fábrica para criar uma classe que agrupa uma nova operação ExecuteTPUEmbeddingPartitioner.
Parâmetros
escopo | escopo atual |
---|---|
configuração | Um proto TPUEmbeddingConfiguration serializado em uma string, descrevendo a configuração TPUEmbedding desejada. |
Devoluções
- uma nova instância de ExecuteTPUEmbeddingPartitioner