fix bug

microsoft · Oct 2, 2024 · 2895f37 · 2895f37
1 parent cb55d4b
commit 2895f37
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc b/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc
@@ -170,9 +170,9 @@ Status LayerNormImpl::ComputeImpl(OpKernelContext* p_ctx, int64_t orig_axis, flo
   }
 
   int output_index = 1;
-  Tensor* mean = p_ctx->Output(output_index++, TensorShape(mean_inv_std_dev_dim));
   U* mean_data = nullptr;
-  if (mean != nullptr) {
+  if (!simplified) {
+    Tensor* mean = p_ctx->Output(output_index++, TensorShape(mean_inv_std_dev_dim));
     mean_data = mean->MutableData<U>();
   }