official/nlp/transformer/embedding_layer.py · 567bd18d4e6c1e31e4e58d3ffee0127f81bc1ab8 · ModelZoo / ResNet50_tensorflow · GitLab

Find file Blame History Permalink

Explicit set the dtype in embedding layer. Otherwise, using the layer in TF1.x... · 567bd18d
A. Unique TensorFlower authored Jul 21, 2020
```
Explicit set the dtype in embedding layer. Otherwise, using the layer in TF1.x will create an int32 tensor.

PiperOrigin-RevId: 322473236
```
567bd18d

embedding_layer.py 3.78 KB