Add support for int32_t indices in TBE training (2G/N)
#10970
| Job | Run time |
|---|---|
| 18m 16s | |
| 20m 34s | |
| 17m 36s | |
| 17m 48s | |
| 18m 29s | |
| 17m 49s | |
| 9m 54s | |
| 9m 17s | |
| 12m 32s | |
| 12m 38s | |
| 2h 34m 53s |