TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

onnx_graphsurgeon: fix Sequence Tensor bug

Open inisis opened this issue 2 years ago • 3 comments

Hi, when using graphsurgeon to handle model with SequenceConstruct operator, constant folding will raise an error, this pr helps fix it, model is provided model.zip.

current output after constant folding looks like this: image

expected output should look like this: image

inisis avatar Jan 24 '24 09:01 inisis

@pranavm-nvidia please review

inisis avatar Jan 24 '24 15:01 inisis

@rajeevsrao Hi, can this be merged.

inisis avatar Jan 25 '24 02:01 inisis

@pranavm-nvidia @rajeevsrao Hi, is there any update? This pr really helps.

inisis avatar Jan 30 '24 12:01 inisis