Forward input dims in Reshape ShapeHelper

[tungld]

In this reshape op in a quantized bert model, the first two dimensions of the output are the same as the ones in the input, so we would like to forward IndexExprs of these dimensions from the input to the output so that DimAnalysis works. But it failed because of using IndexExpr.selectOrSelf.

%184 = onnx.Constant dense<[0, 0, 12, 64]> : tensor<4xi64>
%494 = "onnx.Reshape"(%491, %184) {allowzero = 0 : si64, onnx_node_name = "/bert/encoder/layer.0/attention/self/Reshape"} : (tensor<?x?x768xf32>, tensor<4xi64>) -> tensor<?x?x12x64xf32>

This patch updates selectOrSelf to forward the input QuestionMark when the condition is literal.

[tungld]

@AlexandreEichenberger if you know how to modify selectOrSelf to fix this issue, please let me know or please feel free to create a PR for that. Thanks!

[tungld]

I changed this to make the compare expression literal.

[tungld]

It looks like I found the way to change selectOrSelf but it only works if the condition is literal at compile time.

[AlexandreEichenberger]

@tungld is there still a case to look into?

[tungld]

@tungld is there still a case to look into?

I changed selectOrSelf to handle the case where the comparison is literal. It works for the quantized bert model. I hope it's enough for most cases.

[tungld]

@AlexandreEichenberger It's ready for review.

[AlexandreEichenberger]

LGTM, make sense. Not 100% sure why I added a deepCopy at that time, it makes sense not to generate a new question mark when we have questions marks.

[jenkins-droid]

Jenkins Linux amd64 Build #14969 [push] Forward input dims in Re... started at 10:54

[jenkins-droid]

Jenkins Linux s390x Build #14974 [push] Forward input dims in Re... started at 11:54

[jenkins-droid]

Jenkins Linux ppc64le Build #13999 [push] Forward input dims in Re... started at 12:03

[jenkins-droid]

Jenkins Linux amd64 Build #14969 [push] Forward input dims in Re... passed after 1 hr 7 min

[jenkins-droid]

Jenkins Linux s390x Build #14974 [push] Forward input dims in Re... passed after 1 hr 31 min

[jenkins-droid]

Jenkins Linux ppc64le Build #13999 [push] Forward input dims in Re... passed after 1 hr 59 min