commit b6533bf11263c889ddf835006131945d5847940f
author Patrice Vignola <vignola.patrice@gmail.com> Mon Dec 20 12:06:12 2021 -0800
committer Patrice Vignola <vignola.patrice@gmail.com> Mon Dec 20 12:06:12 2021 -0800
tree 3cdcb7b3a8855fb028578039c50c3c00aab69edf
parent fe72785be081448fc25004f35f0d258416f83bb1

Fixes

tensorflow/core/kernels/mkl/mkl_layer_norm_op.cc

diff --git a/tensorflow/core/kernels/mkl/mkl_layer_norm_op.cc b/tensorflow/core/kernels/mkl/mkl_layer_norm_op.cc
index 9696d70..96cd7d7 100644
--- a/tensorflow/core/kernels/mkl/mkl_layer_norm_op.cc
+++ b/tensorflow/core/kernels/mkl/mkl_layer_norm_op.cc
@@ -118,11 +118,6 @@
                                    MklDnnType<float>(),
                                    memory::format_tag::x},
                                   cpu_engine, shift_buf_dst);
-      void* shift_buf_dst =
-          scale_shift_buf + sizeof(float) * num_elements_scale;
-      auto shift_mem_dst = memory(
-          {{num_elements_shift}, MklDnnType<float>(), memory::format_tag::x},
-          cpu_engine, shift_buf_dst);
       auto shift_reorder_prim = reorder(shift_mem_src, shift_mem_dst);
       std::unordered_map<int, memory> shift_reorder_args;
       shift_reorder_args.insert({DNNL_ARG_FROM, shift_mem_src});