Add support for int32_t indices in TBE training (2H/N)
#8752
| Job | Run time |
|---|---|
| 3s | |
| 19m 37s | |
| 1h 36m 14s | |
| 1h 57m 37s | |
| 1h 59m 0s | |
| 2h 58m 34s | |
| 3h 0m 9s | |
| 21s | |
| 55s | |
| 19s | |
| 49s | |
| 38s | |
| 15s | |
| 11h 54m 31s |