xla

15a8dc62 - Add support for pooling tensors without batch