Add support for int32_t
indices in TBE training (2B/N) (#3371)
#1978
Matrix: build_artifact
Matrix: test_artifact