Add support for int32_t indices in TBE training (2K/N) (#3583)
#8781
| Job | Run time |
|---|---|
| 6s | |
| 6m 4s | |
| 5m 49s | |
| 6m 4s | |
| 5m 54s | |
| 6m 4s | |
| 5m 51s | |
| 6m 8s | |
| 5m 53s | |
| 6m 5s | |
| 5m 51s | |
| 21s | |
| 25s | |
| 9s | |
| 9s | |
| 23s | |
| 10s | |
| 16s | |
| 15s | |
| 22s | |
| 17s | |
| 1h 2m 36s |