xla

fe34a276 - Add max_pool{2,3}d{_with_indices_backward}, mean, sum