Skip to content

Commit 11158e6

Browse files
authored
[alt] fix for issue with input generation (#361)
In <776822a>, a line of code was erroneously removed, resulting in a failure to get the input shape for static dims. This adds back in the missing line to append the static shapes to the int_dims list.
1 parent 776822a commit 11158e6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

alt_e2eshark/e2e_testing/onnx_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ def get_node_shape_from_dim_param_dict(node: onnxruntime.capi.onnxruntime_pybind
4545
raise ValueError(
4646
f"input node '{node.name}' has a non-positive dim: {dim}. Consider setting cutsom inputs for this test."
4747
)
48+
int_dims.append(dim)
4849
return int_dims
4950

5051

0 commit comments

Comments
 (0)