Add support for int32_t indices in TBE training (2G/N)
#8538
| Job | Run time |
|---|---|
| 9s | |
| 20m 57s | |
| 1h 33m 19s | |
| 1h 51m 34s | |
| 1h 51m 39s | |
| 2h 7m 38s | |
| 2h 17m 20s | |
| 20s | |
| 37s | |
| 45s | |
| 33s | |
| 20s | |
| 22s | |
| 10h 5m 33s |