zjgarvey c0eb6d89c0 [ONNX] add some args to the onnx importer to assist shape_inference (#3445) ... Adds the following arguments: - "--clear-domain": enabling this flag (default False) will delete the domain attribute from each node in the onnx model before importing. Shape inference does not seem to work for onnx ops in custom domains. In the rare case when these ops have a corresponding counterpart in base onnx, enabling this flag might allow shape inference to work properly. - "--opset-version": allows setting the opset version manually. This will cause the importer to attempt to update the opset_version of the onnx model before importing. Newer opset versions sometimes have more robust shape inference patterns.		2024-06-12 10:55:14 -05:00
..
torch_mlir	[ONNX] add some args to the onnx importer to assist shape_inference (#3445)	2024-06-12 10:55:14 -05:00
CMakeLists.txt	Extract the Python APIs in the pt1 dir back to the root (#3237)	2024-04-27 18:27:37 +08:00
TorchMLIRModule.cpp	Upstream the ONNX importer. (#2636)	2023-12-12 19:02:51 -08:00