diff --git a/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc b/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc index 2fb53e04f23e..71dd5ab80326 100644 --- a/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc +++ b/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc @@ -173,7 +173,9 @@ Status LayerNormImpl::ComputeImpl(OpKernelContext* p_ctx, int64_t orig_axis, flo U* mean_data = nullptr; if (!simplified) { Tensor* mean = p_ctx->Output(output_index++, TensorShape(mean_inv_std_dev_dim)); - mean_data = mean->MutableData(); + if (mean != nullptr) { + mean_data = mean->MutableData(); + } } U* inv_std_dev_data = nullptr;