Making ids unique in nn.embedding_lookup_sparse. This helps to reduce RPC...
Making ids unique in nn.embedding_lookup_sparse. This helps to reduce RPC calls for looking up the embeddings when there are repeated ids in the batch. PiperOrigin-RevId: 195002785
Loading
Please sign in to comment