Add support for int32_t indices in TBE training (2G/N)
#10954
| Job | Run time |
|---|---|
| 9m 55s | |
| 18m 29s | |
| 9m 22s | |
| 14m 0s | |
| 17m 36s | |
| 14m 20s | |
| 18m 33s | |
| 17m 49s | |
| 11m 34s | |
| 12m 1s | |
| 2h 23m 39s |