Enable cross entropy loss for xla autocast with FP32 precision (#7992… #10103
Job | Run time |
---|---|
7s | |
5m 25s | |
24m 33s | |
23m 55s | |
7m 58s | |
14m 11s | |
4m 38s | |
6m 52s | |
0s | |
1m 36s | |
10m 15s | |
13m 35s | |
4m 55s | |
37m 33s | |
5m 47s | |
9m 30s | |
45m 54s | |
29m 50s | |
10m 53s | |
18m 14s | |
8m 6s | |
6m 26s | |
4h 50m 13s |