xla

abb07e41 - Add nn.functional.adaptive_avg_pool2d (#8010)