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 |
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