-
Notifications
You must be signed in to change notification settings - Fork 97
Open
Description
- Create an eager graph building mode
- The mode needs to be eager, aka the graph is built on the fly, instead of built lazily (?)
- Types for ONNX Script functions should pass mypy checks. This is important for adoption in other code bases because we don't want our users to disable their checks for onnx script
- We can create overloaded methods to handle numpy (concrete value) inputs. The inputs can also be ir.Tensor. Users should annotate the functions to take ir.Value or a derivative of this class (aka. onnxscript Tensor)
- Typing annotations should be simplified because parsing annotations and keeping up with different python versions is a pain. Instead, decorate the functions with onnxscript decorators and declare type information there, if needed.
Metadata
Metadata
Assignees
Labels
topic: discussionFor discussionFor discussion