ExecuteTPUEmbeddingPartitioner

classe final pública ExecuteTPUEmbeddingPartitioner

Um op que executa o particionador TPUEmbedding na configuração central

dispositivo e calcula o tamanho HBM (em bytes) necessário para a operação TPUEmbedding.

Métodos públicos

Saída <String>
comoSaída ()
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.
estático ExecuteTPUEmbeddingPartitioner
criar ( escopo do escopo, configuração de string)
Método de fábrica para criar uma classe que envolve uma nova operação ExecuteTPUEmbeddingPartitioner.

Métodos Herdados

Métodos públicos

Public Output <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.

Public Output <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 de escopo, configuração de string)

Método de fábrica para criar uma classe que envolve uma nova operação ExecuteTPUEmbeddingPartitioner.

Parâmetros
alcance escopo atual
configuração Um proto TPUEmbeddingConfiguration serializado para uma string, descrevendo a configuração TPUEmbedding desejada.
Devoluções
  • uma nova instância de ExecuteTPUEmbeddingPartitioner