Created by: grygielski

This PR adds support for other axes in MKLDNN softmax op. These changes will allow to use MKLDNN softmax op in wider range of models.