XlaRecvTPUEmbeddingActivations

classe final pública XlaRecvTPUEmbeddingActivations

Um op que recebe ativações de incorporação na TPU.

O sistema TPU executa as pesquisas e agregações de incorporação. Os resultados dessas agregações são visíveis para o Tensorflow Graph como as saídas de uma Op XlaRecvTPUEmbeddingActivations. Esta op retorna uma lista contendo um Tensor de ativações por tabela especificada no modelo.

Métodos públicos

static XlaRecvTPUEmbeddingAtivações
create ( Escopo do escopo, Operando <?> deduplicationData, Long numTables, String config)
Método de fábrica para criar uma classe envolvendo uma nova operação XlaRecvTPUEmbeddingActivations.
Iterador< Operando <Float>>
Lista< Saída <Float>>
saídas ()
Uma TensorList de ativações de incorporação contendo um Tensor por tabela de incorporação no modelo.

Métodos Herdados

Métodos públicos

public static XlaRecvTPUEmbeddingActivations create ( Escopo do escopo, Operando <?> deduplicationData, Long numTables, String config)

Método de fábrica para criar uma classe envolvendo uma nova operação XlaRecvTPUEmbeddingActivations.

Parâmetros
alcance escopo atual
dados de desduplicação Um tensor com type=DT_VARIANT contendo os dados de desduplicação. O tensor é uma tupla aninhada XLA contendo N elementos (onde N é a razão do número de incorporação para núcleos de tensor por chip TPU). Cada elemento da tupla aninhada é uma tupla de tensores de nível 1. Cada tensor contém índices (DT_UINT32) para pesquisa de incorporação no TensorCore ou pesos (DT_FLOAT) para aplicar à saída da operação de pesquisa de incorporação.
numTables O número de tensores de ativação de saída. Se o descritor de recurso estiver presente na configuração de incorporação de tpu, ele será igual ao número de recursos, caso contrário, igual ao número de tabelas de incorporação no modelo.
configuração Protocolo serializado de configuração TPUEmbedding.
Devoluções
  • uma nova instância de XlaRecvTPUEmbeddingActivations

public Iterator< Operando <Float>> iterador ()

public List< Output <Float>> outputs ()

Uma TensorList de ativações de incorporação contendo um Tensor por tabela de incorporação no modelo.