diff --git a/kernels/portable/cpu/util/normalization_ops_util.cpp b/kernels/portable/cpu/util/normalization_ops_util.cpp index f711825789..4adcf02b30 100644 --- a/kernels/portable/cpu/util/normalization_ops_util.cpp +++ b/kernels/portable/cpu/util/normalization_ops_util.cpp @@ -38,7 +38,7 @@ bool check_batch_norm_args( ET_LOG_AND_RETURN_IF_FALSE( tensors_have_same_dtype(in, running_mean.value())); } - if (running_mean.has_value()) { + if (running_var.has_value()) { ET_LOG_AND_RETURN_IF_FALSE( tensors_have_same_dtype(in, running_var.value())); }