ONNX shape inference does not infer shapes

Bug Report

Describe the bug

onnx.shape_inference.infer_shapes does not correctly infer shape of each layer.

System information

• OS Platform and Distribution: Windows 10
• ONNX version: 1.7.0
• Python version: 3.7.4

Reproduction instructions

• Describe the code to reproduce the behavior.

model = onnx.load("models/conv_dummy.onnx")
onnx.checker.check_model(model)
inferred_model = onnx.shape_inference.infer_shapes(model)
print(inferred_model.graph.value_info)

output:

[name: "9"
type {
  tensor_type {
    elem_type: 1
  }
}
, name: "10"
type {
  tensor_type {
    elem_type: 1
  }
}
, name: "11"
type {
  tensor_type {
    elem_type: 1
  }
}
, name: "12"
type {
  tensor_type {
    elem_type: 1
  }
}
, name: "13"
type {
  tensor_type {
    elem_type: 1
  }
}
, name: "14"
type {
  tensor_type {
    elem_type: 1
  }
}
]

Model file: models.zip

Expected behavior

Expected each entry in model.graph.value_info to have tensor shape field which tells me the shape of that layer.

Notes

Model was exported from PyTorch using torch.onnx.export

[dtch1997]

Good workaround. Adding a line add_value_info_for_constants(model) before shape inference runs correctly.

[jcwchen]

Actually the utilities here come from protobuf since it’s a model proto. Perhaps you can raise this concern there. Thank you for the suggestion.