Merge pull request #202 from Abioy/master

Improve `DataTypeError` message with `parametrized dimension`
webonnx · Mar 5, 2024 · 7880ed8 · 7880ed8
2 parents acfb490 + cd2f982
commit 7880ed8
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())