graph: backend: dnnl: layout prop: reorder src to abc for LayerNorm

Author: rongzha1

src/graph/backend/dnnl/layout_propagator.cpp

@@ -1,5 +1,5 @@
 /*******************************************************************************
- * Copyright 2022-2023 Intel Corporation
+ * Copyright 2022-2024 Intel Corporation
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -735,6 +735,12 @@ status_t layout_propagator_for_layernorm(op_ptr &op,
     const auto &pd
             = layernorm_executable_t::create_desc(op, p_engine, mgr, pd_cache);
 
+    insert_reorder_before(
+            op, 0, pd.src_desc(), p_engine, mgr, pd_cache, rewriter);
+    value_ptr src = op->get_input_value(0);
+    status = fill_layout_info(src, pd.src_desc());
+    if (status != status::success) return status;
+
     insert_reorder_after(
             op, 0, pd.dst_desc(), p_engine, mgr, pd_cache, rewriter);
     value_ptr dst = op->get_output_value(0);

src/graph/backend/dnnl/op_executable.cpp

@@ -751,9 +751,12 @@ layernorm_executable_t::desc_t layernorm_executable_t::create_desc(
 
     auto src = make_dnnl_memory_desc(
             op->get_input_value(0)->get_logical_tensor());
+    // onednn spec: Implementations optimized for memory formats ab, abc, bac
+    // and abcd
+    src = to_ncx_format(src);
     auto dst = make_dnnl_memory_desc(
             op->get_output_value(0)->get_logical_tensor());
-
+    dst = to_format_any(dst);
     dnnl::layer_normalization_forward::primitive_desc pd(
             p_engine, pkind, src, dst, epsilon, flags, prm_attr);