Address comments. Improve diagnostics in GetImmediateShapeTensor()

nod-ai · Sep 23, 2024 · 62e0867 · 62e0867
1 parent ffa6f2e
commit 62e0867
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/onnxruntime/core/providers/iree/compiler/torch-mlir-import-onnx/OnnxImporter.cpp b/onnxruntime/core/providers/iree/compiler/torch-mlir-import-onnx/OnnxImporter.cpp
@@ -1015,8 +1015,11 @@ Status NodeImporter::GetImmediateShapeTensor(const std::string &name,
   const onnx::TensorProto *tensor =
       graph_info_.graph_viewer().GetConstantInitializer(name, false);
   if (!tensor) {
-    return SetError(
-        "Could not find immediate shape tensor in graph initializers");
+    std::string msg = "Could not find the immediate shape tensor ";
+    msg.append(name);
+    msg.append(" in constant graph initializers. It was possibly produced "
+               "dynamically.");
+    return SetError(msg);
   }
   const onnx::TensorProto &tp = *tensor;