mirror of https://github.com/llvm/torch-mlir
c0eb6d89c0
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. |
||
|---|---|---|
| .. | ||
| _mlir_libs | ||
| dialects | ||
| extras | ||
| tools/import_onnx | ||
| compiler_utils.py | ||
| fx.py | ||