fix error messages

microsoft · Oct 12, 2023 · 1cd4672 · 1cd4672
1 parent d0daf4e
commit 1cd4672
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/core/providers/cpu/nn/instance_norm_helper.h b/onnxruntime/core/providers/cpu/nn/instance_norm_helper.h
@@ -34,7 +34,7 @@ class InstanceNormHelper {
     if (scale->Shape().Size() != in_channels) {
       std::ostringstream ostr;
       ostr << "Mismatch between input data and scale: size of scale != input channel count " << scale->Shape().Size()
-           << " vs. " << input->Shape().GetDims()[1];
+           << " vs. " << in_channels;
       return common::Status(common::ONNXRUNTIME, common::INVALID_ARGUMENT, ostr.str());
     }
 
@@ -47,7 +47,7 @@ class InstanceNormHelper {
     if (B->Shape().Size() != in_channels) {
       std::ostringstream ostr;
       ostr << "Mismatch between input data and B: size of B != input channel count " << B->Shape().Size() << " vs. "
-           << input->Shape().GetDims()[1];
+           << in_channels;
       return common::Status(common::ONNXRUNTIME, common::INVALID_ARGUMENT, ostr.str());
     }