Add support for int32_t indices in TBE training (2G/N)
#8555
| Job | Run time |
|---|---|
| 6s | |
| 20m 42s | |
| 1h 32m 2s | |
| 1h 52m 14s | |
| 1h 53m 4s | |
| 2h 12m 3s | |
| 2h 22m 38s | |
| 15s | |
| 18s | |
| 17s | |
| 18s | |
| 14s | |
| 23s | |
| 10h 14m 34s |