Fix: Align nn.Embedding output with AMP autocast precision
#958
| Job | Run time |
|---|---|
| 1s | |
| 1s | |
| 1s | |
| 3s |
nn.Embedding output with AMP autocast precision
#958
| Job | Run time |
|---|---|
| 1s | |
| 1s | |
| 1s | |
| 3s |