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

[JackCaoG]

This is to fix #2585, in pytorch pr pytorch/pytorch#44790, any and all can take non-boolean input. With our current lowering, reduce won't do anything if all reduce dimensions of the input all have size 1.

(Pdb) torch.all(torch.tensor([-147.5]).to(device))
tensor(-147.5000, device='xla:0')

Add a xla::Select when all reduce dimensions have size 1 to force the result value to be 1 or 0

[davidel]

LGTM with using xla::Zeros()

[JackCaoG]

Can not repo the TestNNDeviceTypeXLA.test_GroupNorm_empty_xla failure, will take another look if it failed again.

[JackCaoG]

Test failure might be due to the pinned pytorch_pr needs a rebase. I manually applied the patch on the pytorch head and run all test with this pr and all test passed.

[JackCaoG]

XLA::NE will change the result type to PRED but torch.any and torch.all expect the result to be the same type as the input. Printing the result of torch.any/all with this implementation will cause error (test performs torch.any().cpu() so it did not catch this).

Use the xla::select instead to fix this.

[JackCaoG]

All test passed locally