xla

0ca733b6 - Add back the implementation of adaptive_pool_2d (#8526)