xla

Add shape fn to adaptive_average_pool2d/backward +log_softmax

#3727

Merged