xla

10db402a - Add shape fn to adaptive_average_pool2d/backward +log_softmax (#3727)