Closed inisis closed 1 year ago
@inisis this is essentially what polygraphy surgeon sanitize --fold-constants
does. There's also an equivalent Python API documented here
@inisis this is essentially what
polygraphy surgeon sanitize --fold-constants
does. There's also an equivalent Python API documented here
@pranavm-nvidia That's so great! How about add this in onnx-graphsurgeon.
closing since this is al ready solved, thanks all!
Hi, when shape info is missing within a onnx model. calling fold_constants will fail for many cases, so I wonder if shape_infernce can be added before import_graph, a simple case can be find here test.zip
and a quick fix is as below, it's really hard to insert shape inference within graphsurgeon.