TODO: Emit the subfunction in block if cond

microsoft · Oct 26, 2023 · 2ec39c5 · 2ec39c5
1 parent 4d7ac4d
commit 2ec39c5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/onnxscript/irbuilder.py b/onnxscript/irbuilder.py
@@ -361,6 +361,8 @@ def to_proto(f):
                 if isinstance(f, onnx.FunctionProto):
                     return f
                 if isinstance(f, onnxscript.OnnxFunction):
+                    # TODO(justinchuby): functions should be function protos
+                    # Normalize the OnnxFunction instead?
                     return f.to_function_proto()
                 raise TypeError("Expected a value of type FunctionProto of OnnxFunction")