[Stream] Enable batch affinity queries in SpecializeEncoding pass. #34664

Job	Run time
setup / setup	8s
runtime_small	3m 57s
runtime :: ubuntu-24.04	1m 21s
runtime_tracing :: ubuntu-24.04 :: tracy	1m 13s
runtime :: windows-2022	2m 44s
runtime_tracing :: ubuntu-24.04 :: console	1m 5s
runtime :: macos-14	1m 57s
runtime_tracing :: windows-2022 :: tracy	2m 14s
runtime_tracing :: windows-2022 :: console	2m 21s
runtime_tracing :: macos-14 :: tracy	1m 3s
runtime_tracing :: macos-14 :: console	1m 5s
linux_x64_bazel / linux_x64_bazel	30m 56s
linux_x64_clang / linux_x64_clang	30m 26s
linux_x64_clang_asan / linux_x64_clang_asan	21m 34s
ci_summary / summary	1s
	1h 42m 5s

Provide feedback