It would be much easier to have that than emulate using arithmetics, which doesn't work well, e.g., #48351

cc @jlin27 @mruberry