ConfigureTPUEmbeddingMemory

classe final pública ConfigureTPUEmbeddingMemory

Uma operação que configura o software TPUEmbedding em um host.

Métodos Públicos

Saída <String>
asOutput ()
Retorna o identificador simbólico de um tensor.
ConfigureTPUEmbeddingMemory estático
create ( Escopo , Operando <String> commonConfig)
Método de fábrica para criar uma classe que agrupa uma nova operação ConfigureTPUEmbeddingMemory.
Saída <String>
memóriaConfig ()
Uma configuração de memória codificada em string contendo metadados sobre as alocações de memória reservadas para TPUEmbedding.

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.

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

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

Parâmetros
escopo escopo atual
commonConfig Um proto CommonConfiguration codificado em string contendo metadados sobre a saída do particionador TPUEmbedding e o tamanho do HBM (em bytes) necessário para a operação.
Devoluções
  • uma nova instância de ConfigureTPUEmbeddingMemory

Saída pública <String> memoryConfig ()

Uma configuração de memória codificada em string contendo metadados sobre as alocações de memória reservadas para TPUEmbedding.