-
Notifications
You must be signed in to change notification settings - Fork 3.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Relay] add python doc for function in ir_pass #1877
Conversation
Weird, I thought the env already was the second argument for |
@slyubomirsky it was probably lost in the sea of git branches. |
It is all good. |
If the tests pass, it looks good to me. |
python/tvm/relay/ir_pass.py
Outdated
t: relay.Type | ||
The type to check | ||
|
||
env: relay.Environment |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
env: relay.Environment, optional
@tqchen I had addressed the review comment. |
It is almost done - I have no idea what to do with check_kind.
If there is only 1 arg, it is interpreted as type.
If there is only 2 arg, the first is env, and the second is type.
what should I do? redesign the check_kind interface to has env coming last?