xla

825ba0da - Move op dispatching logic into an `Environment` class; and use Mode to capture dispatcher instead of tensor. (#7009)