x64: lrn, softmax: move setting formats before post_ops_ok check

Author: tczeszun

src/cpu/x64/jit_uni_layer_normalization.cpp

@@ -1233,9 +1233,9 @@ status_t jit_uni_layer_normalization_fwd_t::pd_t::init(engine_t *engine) {
 
         return injector::post_ops_ok(post_ops_args);
     };
-    VDISPATCH_LNORM(post_ops_ok(), VERBOSE_UNSUPPORTED_POSTOP);
     VDISPATCH_LNORM(attr_.set_default_formats(dst_md(0)) == status::success,
             VERBOSE_UNSUPPORTED_POSTOP);
+    VDISPATCH_LNORM(post_ops_ok(), VERBOSE_UNSUPPORTED_POSTOP);
 
     VDISPATCH_LNORM(fill_compatible_stats_md(*src_md(), reordered_stat_md_)
                     == status::success,

src/cpu/x64/jit_uni_softmax.hpp

@@ -135,13 +135,13 @@ struct jit_uni_softmax_fwd_t : public primitive_t {
                             | skip_mask_t::post_ops),
                     VERBOSE_UNSUPPORTED_ATTR);
             VDISPATCH_SOFTMAX(attr_scales_ok(), VERBOSE_UNSUPPORTED_SCALES_CFG);
-            VDISPATCH_SOFTMAX(post_ops_ok(), VERBOSE_UNSUPPORTED_POSTOP);
 
             VDISPATCH_SOFTMAX(set_default_formats() == status::success,
                     VERBOSE_UNSUPPORTED_TAG);
             VDISPATCH_SOFTMAX(
                     attr_.set_default_formats(dst_md(0)) == status::success,
                     VERBOSE_UNSUPPORTED_TAG);
+            VDISPATCH_SOFTMAX(post_ops_ok(), VERBOSE_UNSUPPORTED_POSTOP);
             VDISPATCH_SOFTMAX(
                     memory_desc_wrapper(src_md()).similar_to(
                             memory_desc_wrapper(dst_md()), true, false, 0),