Add support for int32_t indices in TBE training (2H/N)
#7490
| Job | Run time |
|---|---|
| 27m 10s | |
| 27m 58s | |
| 27m 37s | |
| 28m 19s | |
| 27m 5s | |
| 27m 28s | |
| 27m 42s | |
| 28m 13s | |
| 27m 44s | |
| 27m 24s | |
| 27m 26s | |
| 27m 35s | |
| 27m 1s | |
| 26m 57s | |
| 27m 51s | |
| 27m 53s | |
| 26m 55s | |
| 27m 40s | |
| 27m 43s | |
| 28m 24s | |
| 23m 16s | |
| 23m 1s | |
| 28m 2s | |
| 28m 10s | |
| 10h 54m 34s |