tensorflow-onnx icon indicating copy to clipboard operation
tensorflow-onnx copied to clipboard

Published 20 hours ago • verified publisher icon onnx

Add shape inference as an arg for tools/onnx-optimize.py

Open hwangdeyu opened this issue 2 years ago • 1 comments

In https://github.com/onnx/tensorflow-onnx/blob/1915fb52270856debc74ba5d28e332e46ccc1e1a/tools/onnx-optimize.py#L59 the optimizer.optimize_graph() method would eliminate infered shape. This PR adds the shape inference after model optimized.

Signed-off-by: Deyu Huang [email protected]

hwangdeyu avatar Jun 08 '22 12:06 hwangdeyu

An example node change. example

hwangdeyu avatar Jun 08 '22 12:06 hwangdeyu