Add support for int32_t
indices in TBE training (2D/N) (#3374)
#1996
Matrix: build_artifact
Matrix: test_artifact