xla

Fix issue with any/all when all reduce dimensions of input have size 1

#2590

Merged