Add onnx to InputType enum (iree-org#18310)

Add onnx to `InputType` enum. Signed-off-by: Tim Gymnich <[email protected]>
nod-ai · Aug 21, 2024 · 6a92fb7 · 6a92fb7
1 parent 7c8fedc
commit 6a92fb7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/compiler/bindings/python/iree/compiler/tools/core.py b/compiler/bindings/python/iree/compiler/tools/core.py
@@ -48,6 +48,7 @@ class InputType(Enum):
     TOSA = "tosa"
     TM_TENSOR = "tm_tensor"
     TORCH = "torch"
+    ONNX = "onnx"
 
     @staticmethod
     def parse(spec: Union[str, InputType]) -> InputType: