ExecuteTPUEmbeddingPartitioner

classe final pública ExecuteTPUEmbeddingPartitioner

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
create (escopo do escopo , configuração de string)
Método de fábrica para criar uma classe que agrupa uma nova operação ExecuteTPUEmbeddingPartitioner.

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