Improve DataTypeError message with parametrized dimension

webonnx · Mar 5, 2024 · cd2f982 · cd2f982
1 parent acfb490
commit cd2f982
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/wonnx/src/utils.rs b/wonnx/src/utils.rs
@@ -487,10 +487,11 @@ impl ValueInfoProto {
                         .get_shape()
                         .get_dim()
                         .iter()
-                        .map(|x| {
+                        .enumerate()
+                        .map(|(idx, x)| {
                             if x.has_dim_param() {
                                 return Err(DataTypeError::ParametrizedDimensionUnsupported(
-                                    x.get_dim_param().to_string(),
+                                    self.get_name().to_string() + "#" + &idx.to_string(),
                                 ));
                             }
                             Ok(x.get_dim_value())