xla

7d681a99 - Improve assume_pure docs and tests (#9038)