official/nlp/transformer/embedding_layer.py · 1a8fd30ec2fa1dabb14831a5de4658209afbf667 · ModelZoo / ResNet50_tensorflow · GitLab

Find file Blame History Permalink

Explicit set the dtype in embedding layer. Otherwise, using the layer in TF1.x... · 1a8fd30e
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
```
1a8fd30e

embedding_layer.py 3.78 KB