View source on GitHub |
Embedding feature flag strings.
UNSUPPORTED: No embedding lookup accelerator available on the tpu. V1: Embedding lookup accelerator V1. The embedding lookup operation can only be placed at the beginning of computation. Only one instance of embedding lookup layer is allowed. V2: Embedding lookup accelerator V2. The embedding lookup operation can be placed anywhere of the computation. Multiple instances of embedding lookup layer is allowed.
Class Variables | |
---|---|
UNSUPPORTED |
<EmbeddingFeature.UNSUPPORTED: 'UNSUPPORTED'>
|
V1 |
<EmbeddingFeature.V1: 'V1'>
|
V2 |
<EmbeddingFeature.V2: 'V2'>
|