Skip to content

Commit

Permalink
Performance enhancements for sparse embedding lookups.
Browse files Browse the repository at this point in the history
  • Loading branch information
philipphack committed Feb 16, 2023
1 parent 2166e5d commit 13dd2c8
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions RELEASE.md
Expand Up @@ -34,13 +34,14 @@
model and exports it if the user-defined comparison function determines
that it is an improvement.

* Embeddings:
* `tf.nn`

* `tf.nn.embedding_lookup_sparse` and `tf.nn.safe_embedding_lookup_sparse`
now support ids and weights described by RaggedTensors, which are more
memory efficient than SparseTensors. The new boolean argument
`allow_fast_lookup` enables a simplified and typically faster lookup
procedure.
now support ids and weights described by `tf.RaggedTensor`s.
* Added a new boolean argument `allow_fast_lookup` to
`tf.nn.embedding_lookup_sparse` and
`tf.nn.safe_embedding_lookup_sparse`, which enables a simplified and
typically faster lookup procedure.

## Bug Fixes and Other Changes

Expand Down

0 comments on commit 13dd2c8

Please sign in to comment.