Add support for int32_t indices in TBE training (2D/N) (#3374)
#10977
| Job | Run time |
|---|---|
| 9m 54s | |
| 13m 35s | |
| 18m 49s | |
| 9m 14s | |
| 10m 50s | |
| 17m 43s | |
| 12m 13s | |
| 18m 33s | |
| 12m 22s | |
| 17m 48s | |
| 2h 21m 1s |