[beam] Use direct iota instead of binary shift Upstream-commit: ee4fa10fb018b30312dce86592ae6e23f0d588f2 Component: engine