tf.tpu.experimental.HardwareFeature.EmbeddingFeature
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'> |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/tpu/experimental/HardwareFeature/EmbeddingFeature
