Add support for `int32_t` indices in TBE training (2D/N)#3374

Closed

q10 wants to merge 2 commits intopytorch:mainfrom q10:export-D65930273

+17-7

Commits on Jan 10, 2025

Add support for int32_t indices in TBE training (2C/N) (pytorch#3372 )

q10
authored and
facebook-github-bot
committed
Add support for int32_t indices in TBE training (2D/N) (pytorch#3374 )

q10
authored and
facebook-github-bot
committed