Add support for int64_t indices in TBE inference [2/N] #8741

Job	Run time
build-linux (x86, linux.12xlarge, static, gcc)	17m 35s
build-macos (macos-13, static)	15m 53s
build-linux (x86, linux.12xlarge, static, clang)	16m 38s
build-macos (macos-13, shared)	9m 45s
build-linux (x86, linux.12xlarge, shared, gcc)	17m 30s
build-linux (x86, linux.12xlarge, shared, clang)	16m 10s
build-bazel (x86, linux.12xlarge, gcc)	8m 40s
build-bazel (x86, linux.12xlarge, clang)	7m 57s
build-windows (windows-2022, static)	12m 44s
build-windows (windows-2022, shared)	12m 11s
	2h 15m 3s

Provide feedback